Here is the deal. AI makes an AX mag used for actual real AI-AX rifles, they also make an AICS-AX mag. The stock you have is an AICS so you need to use that mag not one made for an AX. An AICS-AW mag will work but not suggested by AI.
This is the mag you need to buy #6955 is the AI part...