How to set version code for Android App Bundle(aab) with flavor dimension?

Question

Setting the version code for aab by overwriting manifest file in gradle in android.tools.build:3.2.1 as below is work fine. After upgrade to android.tools.build:4.0.1, this way