Fixed the bug happened when running the test second time.