diff --git a/azure/durable_functions/__init__.py b/azure/durable_functions/__init__.py
index 7b48cc30..22b77b8d 100644
--- a/azure/durable_functions/__init__.py
+++ b/azure/durable_functions/__init__.py
@@ -6,6 +6,7 @@
 from .entity import Entity
 from .models.utils.entity_utils import EntityId
 from .models.DurableOrchestrationClient import DurableOrchestrationClient
+from .models.OrchestrationRuntimeStatus import OrchestrationRuntimeStatus
 from .models.DurableOrchestrationContext import DurableOrchestrationContext
 from .models.DurableEntityContext import DurableEntityContext
 from .models.RetryOptions import RetryOptions
@@ -67,5 +68,6 @@ def validate_extension_bundles():
     'DurableEntityContext',
     'DurableOrchestrationContext',
     'ManagedIdentityTokenSource',
+    'OrchestrationRuntimeStatus',
     'RetryOptions'
 ]