diff --git a/src/code_.py b/src/code_.py
index f9ea1450508dafcbffb1f498bdc517253c448151..dab97af9f221e608ccf8e92f3e917c3df907825b 100644
--- a/src/code_.py
+++ b/src/code_.py
@@ -106,7 +106,7 @@ class Code:
                                 _, end_condition = match.span(1)
                                 extracted_condition = get_code_until_parenthese_count(code=temp_code[end_condition:], val=-1)
                                 condition = extracted_condition[:-1]
-                                body = extarcted.replace(extracted_condition.strip(), "")
+                                body = extarcted.replace(extracted_condition.strip(), "").strip()
                                 if(body!="" and body[0]!="{"):
                                     new = f"if ({condition}) {{\n{body}\n}}\n"
                                     to_replace.append((all, new))