diff --git a/python/fedml/api/__init__.py b/python/fedml/api/__init__.py
index f753e4255b..cdce1058ba 100755
--- a/python/fedml/api/__init__.py
+++ b/python/fedml/api/__init__.py
@@ -27,6 +27,7 @@
 from fedml.computing.scheduler.model_scheduler.device_server_constants import ServerConstants
 from fedml.computing.scheduler.model_scheduler.device_client_constants import ClientConstants
 
+from .fedml_response import ResponseCode
 
 def fedml_login(api_key: str = None):
     """