Fixed mocking assertions