diff --git a/python/hopsworks/project.py b/python/hopsworks/project.py
index 294a69435..79ccff369 100644
--- a/python/hopsworks/project.py
+++ b/python/hopsworks/project.py
@@ -17,6 +17,7 @@
 
 import json
 
+import hsfs.feature_store
 import humps
 from hopsworks import client, constants, util
 from hopsworks.client.external import Client
@@ -29,7 +30,6 @@
     kafka_api,
     opensearch_api,
 )
-from hsfs import feature_store
 
 
 class Project:
@@ -103,7 +103,9 @@ def created(self):
         """Timestamp when the project was created"""
         return self._created
 
-    def get_feature_store(self, name: str = None, engine: str = None) -> feature_store.FeatureStore:
+    def get_feature_store(
+        self, name: str = None, engine: str = None
+    ) -> hsfs.feature_store.FeatureStore:
         """Connect to Project's Feature Store.
 
         Defaulting to the project name of default feature store. To get a
@@ -142,7 +144,9 @@ def get_feature_store(self, name: str = None, engine: str = None) -> feature_sto
                 engine=engine,
             ).get_feature_store(name)
         else:
-            return connection(engine=engine).get_feature_store(name)  # If internal client
+            return connection(engine=engine).get_feature_store(
+                name
+            )  # If internal client
 
     def get_model_registry(self):
         """Connect to Project's Model Registry API.