How to better control IMA module?