Fixing another failing test case