fixing IMR test failure