Fixed one test case failing