diff --git a/src/server/game/AI/SmartScripts/SmartScript.cpp b/src/server/game/AI/SmartScripts/SmartScript.cpp
index a3605641ae..f8d5600bdd 100644
--- a/src/server/game/AI/SmartScripts/SmartScript.cpp
+++ b/src/server/game/AI/SmartScripts/SmartScript.cpp
@@ -71,7 +71,7 @@ bool SmartScript::IsSmart(Creature* c, bool silent)
         return false;
 
     bool smart = true;
-    if (!me || me->GetAIName() != "SmartAI")
+    if (!dynamic_cast<SmartAI*>(c->AI()))
         smart = false;
 
     if (!smart && !silent)
@@ -87,7 +87,7 @@ bool SmartScript::IsSmart(GameObject* g, bool silent)
         return false;
 
     bool smart = true;
-    if (g && g->GetAIName() != "SmartGameObjectAI")
+    if (!dynamic_cast<SmartGameObjectAI*>(g->AI()))
         smart = false;
 
     if (!smart && !silent)