Fixed import in database_mock.py