Bug fix in test im logic