diff --git a/google/cloud/aiplatform/base.py b/google/cloud/aiplatform/base.py
index b45331a958..dcec7bb5d6 100644
--- a/google/cloud/aiplatform/base.py
+++ b/google/cloud/aiplatform/base.py
@@ -677,7 +677,7 @@ def to_dict(self) -> Dict[str, Any]:
     def _generate_display_name(cls, prefix: Optional[str] = None) -> str:
         """Returns a display name containing class name and time string."""
         if not prefix:
-            prefix = cls.name
+            prefix = cls.__name__
         return prefix + " " + datetime.datetime.now().isoformat(sep=" ")