Fix import in cursor_mock.py