corrected unti test case