Fixed the mock test