diff --git a/app/client/src/entities/Action/index.ts b/app/client/src/entities/Action/index.ts
index bea5f25a537..87d08621d35 100644
--- a/app/client/src/entities/Action/index.ts
+++ b/app/client/src/entities/Action/index.ts
@@ -273,7 +273,7 @@ export function isSaaSAction(action: Action): action is SaaSAction {
   return action.pluginType === PluginType.SAAS;
 }
 
-export function isAIAction(action: Action): action is SaaSAction {
+export function isAIAction(action: Action): action is AIAction {
   return action.pluginType === PluginType.AI;
 }