Had a whoops in mock class