diff --git a/sdk/ml/azure-ai-ml/azure/ai/ml/_telemetry/activity.py b/sdk/ml/azure-ai-ml/azure/ai/ml/_telemetry/activity.py
index a7fae6dc584e..d562f4a4b977 100644
--- a/sdk/ml/azure-ai-ml/azure/ai/ml/_telemetry/activity.py
+++ b/sdk/ml/azure-ai-ml/azure/ai/ml/_telemetry/activity.py
@@ -283,9 +283,12 @@ def wrapper(*args, **kwargs):
                         logger.package_logger, activity_name or f.__name__, activity_type, custom_dimensions
                     ):
                         return f(*args, **kwargs)
-            else:
+            elif hasattr(logger, "package_logger"):
                 with log_activity(logger.package_logger, activity_name or f.__name__, activity_type, custom_dimensions):
                     return f(*args, **kwargs)
+            else:
+                with log_activity(logger, activity_name or f.__name__, activity_type, custom_dimensions):
+                    return f(*args, **kwargs)
 
         return wrapper