Fixing a test case