Fixed the mock tests