Fix the issue where the mock is persisting calls