Fix mocking for production