diff --git a/openhexa/sdk/workspaces/current_workspace.py b/openhexa/sdk/workspaces/current_workspace.py
index 189edbab..a0026512 100644
--- a/openhexa/sdk/workspaces/current_workspace.py
+++ b/openhexa/sdk/workspaces/current_workspace.py
@@ -9,7 +9,6 @@
 
 from openhexa.utils import stringcase
 
-from ...cli.graphql.graphql_client import GetCountriesWorkspaceCountries
 from ..datasets import Dataset
 from ..utils import graphql
 from .connection import (
@@ -61,7 +60,7 @@ def slug(self) -> str:
             raise WorkspaceConfigError("The workspace slug is not available in this environment.")
 
     @property
-    def countries(self) -> list[GetCountriesWorkspaceCountries]:
+    def countries(self):
         """The countries of the workspace."""
         from openhexa.cli.api import OpenHexaClient