diff --git a/Generals/Code/GameEngine/Source/Common/RTS/Team.cpp b/Generals/Code/GameEngine/Source/Common/RTS/Team.cpp index c9016d189ad..0ad97f2f28b 100644 --- a/Generals/Code/GameEngine/Source/Common/RTS/Team.cpp +++ b/Generals/Code/GameEngine/Source/Common/RTS/Team.cpp @@ -364,9 +364,11 @@ Team *TeamFactory::createInactiveTeam(const AsciiString& name) // ------------------------------------------------------------------------ Team *TeamFactory::createTeam(const AsciiString& name) { - Team *t = NULL; - t = createInactiveTeam(name); - t->setActive(); + Team *t = createInactiveTeam(name); + + if (t) + t->setActive(); + return t; } diff --git a/GeneralsMD/Code/GameEngine/Source/Common/RTS/Team.cpp b/GeneralsMD/Code/GameEngine/Source/Common/RTS/Team.cpp index 30c8f077215..eacb40f0900 100644 --- a/GeneralsMD/Code/GameEngine/Source/Common/RTS/Team.cpp +++ b/GeneralsMD/Code/GameEngine/Source/Common/RTS/Team.cpp @@ -364,9 +364,11 @@ Team *TeamFactory::createInactiveTeam(const AsciiString& name) // ------------------------------------------------------------------------ Team *TeamFactory::createTeam(const AsciiString& name) { - Team *t = NULL; - t = createInactiveTeam(name); - t->setActive(); + Team *t = createInactiveTeam(name); + + if (t) + t->setActive(); + return t; }