diff --git a/src/code_.py b/src/code_.py
index cd62a1bb4e494753719fe1908e9310321ed701a0..5296712f62ae89b92343f444ef3009aac6e9d212 100644
--- a/src/code_.py
+++ b/src/code_.py
@@ -45,7 +45,7 @@ class Code:
                 start, end = match.span(1)
                 extracted_condition = get_code_until_parenthese_count(code=code[end:], val=-1)
                 condition = extracted_condition[:-1]
-                body = extarcted.replace(extracted_condition, "")
+                body = extarcted.replace(extracted_condition.strip(), "")
                 new = f"if ({condition}) {{\n{body}\n}}\n"
                 to_replace.append((all, new))
         for r in to_replace: