Fix import using module args