Fixing the test case's bug