Fixed the testcase