Fixing the Test case