1 files changed, 4 insertions, 2 deletions
diff --git a/src/server/game/Maps/Map.h b/src/server/game/Maps/Map.h
index 68a81710c64..8f06fd77538 100644
--- a/src/server/game/Maps/Map.h
+++ b/src/server/game/Maps/Map.h
@@ -65,7 +65,6 @@ class Weather;
 class WorldObject;
 class WorldPacket;
 struct DungeonEncounterEntry;
-struct UpdateSaveDataEvent;
 struct MapDifficultyEntry;
 struct MapEntry;
 struct Position;
@@ -73,6 +72,8 @@ struct ScriptAction;
 struct ScriptInfo;
 struct SmoothPhasingInfo;
 struct SummonPropertiesEntry;
+struct UpdateAdditionalSaveDataEvent;
+struct UpdateBossStateSaveDataEvent;
 class Transport;
 enum Difficulty : uint8;
 enum WeatherState : uint32;
@@ -814,7 +815,8 @@ class TC_GAME_API InstanceMap : public Map
         InstanceScenario const* GetInstanceScenario() const { return i_scenario; }
         void SetInstanceScenario(InstanceScenario* scenario) { i_scenario = scenario; }
         InstanceLock const* GetInstanceLock() const { return i_instanceLock; }
-        void UpdateInstanceLock(DungeonEncounterEntry const* dungeonEncounter, UpdateSaveDataEvent const& updateSaveDataEvent);
+        void UpdateInstanceLock(UpdateBossStateSaveDataEvent const& updateSaveDataEvent);
+        void UpdateInstanceLock(UpdateAdditionalSaveDataEvent const& updateSaveDataEvent);
         void CreateInstanceLockForPlayer(Player* player);
         void UnloadAll() override;
         EnterState CannotEnter(Player* player) override;