fixed a mock in test