diff --git a/src/LuaEngine/methods/UnitMethods.h b/src/LuaEngine/methods/UnitMethods.h
index c23ed33..19ed5b7 100644
--- a/src/LuaEngine/methods/UnitMethods.h
+++ b/src/LuaEngine/methods/UnitMethods.h
@@ -2758,7 +2758,7 @@ namespace LuaUnit
      */
     int IsPet(lua_State* L, Unit* unit)
     {
-        ALE::Push(L, creature->IsPet());
+        ALE::Push(L, unit->IsPet());
         return 1;
     }
 
@@ -2769,7 +2769,7 @@ namespace LuaUnit
      */
     int ToPet(lua_State* L, Unit* unit)
     {
-        ALE::Push(L, creature->ToPet());
+        ALE::Push(L, unit->ToPet());
         return 1;
     }
 };