adding mock in tests