diff --git a/lib/entity_class/entity_wrapper.class.dart b/lib/entity_class/entity_wrapper.class.dart
index 4ea0953..983dc4b 100644
--- a/lib/entity_class/entity_wrapper.class.dart
+++ b/lib/entity_class/entity_wrapper.class.dart
@@ -16,6 +16,9 @@ class EntityWrapper {
   }) {
     this.icon = icon ?? entity.icon;
     this.displayName = displayName ?? entity.displayName;
+    if (this.uiAction == null) {
+      this.uiAction = EntityUIAction();
+    }
   }
 
   void handleTap() {