diff --git a/infrahub_sdk/__init__.py b/infrahub_sdk/__init__.py
index 624f7ee0..e16aedd5 100644
--- a/infrahub_sdk/__init__.py
+++ b/infrahub_sdk/__init__.py
@@ -74,4 +74,7 @@
     "ValidationError",
 ]
 
-__version__ = importlib.metadata.version("infrahub-sdk")
+try:
+    __version__ = importlib.metadata.version("infrahub-sdk")
+except importlib.metadata.PackageNotFoundError:
+    __version__ = importlib.metadata.version("infrahub-server")