diff --git a/tests/conftest.py b/tests/conftest.py
index 6b9cc51..b55602e 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -6,7 +6,7 @@ import pytest
 
 from galaxy.api.plugin import Plugin
 from galaxy.api.consts import Platform
-from tests.async_mock import AsyncMock
+from galaxy.unittest.mock import AsyncMock
 
 @pytest.fixture()
 def reader():