Fixed test for code coverage [Kaizer/Tushar]