fix mock in tests