diff --git a/pandasai/pipelines/chat/code_cleaning.py b/pandasai/pipelines/chat/code_cleaning.py
index a6e047b18..759813b14 100644
--- a/pandasai/pipelines/chat/code_cleaning.py
+++ b/pandasai/pipelines/chat/code_cleaning.py
@@ -176,7 +176,7 @@ def _is_malicious_code(self, code) -> bool:
             "(chr",
             "b64decode",
         ]
-        return any(module in code for module in dangerous_modules)
+        return any(re.search(r'\b'+re.escape(module)+r'\b',code) for module in dangerous_modules)
 
     def _is_jailbreak(self, node: ast.stmt) -> bool:
         """