diff --git a/src/mod_swift_travel_form.cpp b/src/mod_swift_travel_form.cpp
index bf772b9..2b61da9 100644
--- a/src/mod_swift_travel_form.cpp
+++ b/src/mod_swift_travel_form.cpp
@@ -67,7 +67,7 @@ class SwiftTravelForm_UnitScript : public UnitScript
 public:
     SwiftTravelForm_UnitScript() : UnitScript("SwiftTravelForm_UnitScript") {}
 
-    void OnUnitSetShapeshiftForm(Unit* unit, ShapeshiftForm form) override
+    void OnUnitSetShapeshiftForm(Unit* unit, uint8 form) override
     {
         if (!sEnabled)
             return;