Here is a downloadable PS Action that you can import and run before invoking Mask AI. Remember this should just be used to create a Layer Mask from the base layer and pass that to Mask AI. The Action Group is called MaskAI and the action to run is called ReUse Mask.
Just import the action into the action panel in PS.
Download Link: https://www.dropbox.com/s/v4tozwrnpzsd216/MaskAI.atn?dl=0
Note: Created and tested on CS6 Win 10