diff --git a/src/modules/ai/ai.ts b/src/modules/ai/ai.ts
index 702887b..344dd13 100644
--- a/src/modules/ai/ai.ts
+++ b/src/modules/ai/ai.ts
@@ -48,7 +48,7 @@ const onAI = async (
 
     const useSmartModel = text.startsWith("!aipro");
 
-    let textMod = text.replace("!ai", "").trim();
+    let textMod = text.replace("!aipro", "").replace("!ai", "").trim();
     let instructions = {
         prefferedLanguages: ["english", "slovak"],
         users: alts.map((alt) => ({
@@ -105,7 +105,11 @@ const onImageGen = async (text: string, roomId: string, sender: string) => {
 
     const useSmartModel = text.startsWith("!imgpro");
 
-    let textMod = text.replace("!img", "").trim().toLowerCase();
+    let textMod = text
+        .replace("!imgpro", "")
+        .replace("!img", "")
+        .trim()
+        .toLowerCase();
     alts.forEach((alt) => {
         alt.keys.forEach((key) => {
             textMod = textMod.replaceAll(key, alt.alt);