Fixing one more test case