diff --git a/nvflare/client/api.py b/nvflare/client/api.py
index 0e570086d1..87ae416905 100644
--- a/nvflare/client/api.py
+++ b/nvflare/client/api.py
@@ -19,11 +19,9 @@
 from nvflare.apis.analytix import AnalyticsDataType
 from nvflare.app_common.abstract.fl_model import FLModel
 
-from .api_context import APIContext, ClientAPIType
-
 # this import is to let existing scripts import client.api
 from .api_context import ClientAPIType  # noqa: F401
-
+from .api_context import APIContext
 
 global_context_lock = Lock()
 context_dict = {}