Quick fix for successfully running the test case.