fixing a test case which was failing