diff --git a/keepercommander/discovery_common/infrastructure.py b/keepercommander/discovery_common/infrastructure.py
index c396baec8..ee7ad7fed 100644
--- a/keepercommander/discovery_common/infrastructure.py
+++ b/keepercommander/discovery_common/infrastructure.py
@@ -5,7 +5,7 @@
 from ..keeper_dag.exceptions import DAGVertexException
 from ..keeper_dag.crypto import urlsafe_str_to_bytes
 from ..keeper_dag.types import PamGraphId
-from discovery_common.types import DiscoveryObject
+from .types import DiscoveryObject
 import os
 import importlib
 import time