diff --git a/Barotrauma/BarotraumaShared/Lua/PostSetup.lua b/Barotrauma/BarotraumaShared/Lua/PostSetup.lua index ae69bd735..0ac5ec464 100644 --- a/Barotrauma/BarotraumaShared/Lua/PostSetup.lua +++ b/Barotrauma/BarotraumaShared/Lua/PostSetup.lua @@ -23,7 +23,7 @@ end local originalRegisterType = LuaUserData.RegisterType LuaUserData.RegisterType = function (typeName) - if not IsAllowed(typeName) then + if not LuaUserData.IsRegistered(typeName) and not IsAllowed(typeName) then error("Couldn't register type " .. typeName .. ".", 2) end @@ -38,8 +38,8 @@ end local originalCreateStatic = LuaUserData.CreateStatic LuaUserData.CreateStatic = function (typeName, addCallMethod) - if not IsAllowed(typeName) then - error("Couldn't register type " .. typeName .. ".", 2) + if not LuaUserData.IsRegistered(typeName) and not IsAllowed(typeName) then + error("Couldn't create static type " .. typeName .. ".", 2) end local success, result = pcall(originalCreateStatic, typeName, addCallMethod)