correct the error from cleaning up, adding one more test case