diff --git a/ElunaInstanceAI.h b/ElunaInstanceAI.h
index e950eaf..4da8323 100644
--- a/ElunaInstanceAI.h
+++ b/ElunaInstanceAI.h
@@ -14,6 +14,9 @@
 #include "InstanceData.h"
 #endif
 
+#ifdef TRINITY
+#include "Map.h"
+#endif
 
 /*
  * This class is a small wrapper around `InstanceData`,
@@ -61,9 +64,15 @@ private:
     std::string lastSaveData;
 
 public:
+#ifdef TRINITY
+    ElunaInstanceAI(Map* map) : InstanceData(map->ToInstanceMap())
+    {
+    }
+#else
     ElunaInstanceAI(Map* map) : InstanceData(map)
     {
     }
+#endif
 
 #ifndef TRINITY
     void Initialize() override;