@katesfb Apologies for the late answer and you may have gotten this resolved by now. However, I'll respond just in case someone else might have the same question in the future.
From looking into this a bit further, they should be the same device, so yes, technically it should work. Just to be sure though, I recommend confirming this with the contributor of the Mi A2 to ensure compatibility.