Fixed import in mongo_client_mock.py