X-Git-Url: https://code.kerkeslager.com/?p=fur;a=blobdiff_plain;f=normalization.py;h=762a8f1aa39e333e473c3fadbbc925f45377cb33;hp=fbe3d04835eaa753eab4aee4b71c6c29cb498791;hb=02d64def49065ad614fe0ee2a85060666552192e;hpb=151f60b119247efb1bcf05a664f4324b71fac782

diff --git a/normalization.py b/normalization.py
index fbe3d04..762a8f1 100644
--- a/normalization.py
+++ b/normalization.py
@@ -587,12 +587,12 @@ def normalize_statement_list(counter, statement_list, **kwargs):
             result_statement_list.append(s)
         result_statement_list.append(normalized)
 
+    # TODO The way we fix the last statement is really confusing
     last_statement = result_statement_list[-1]
 
     if isinstance(last_statement, NormalExpressionStatement) and isinstance(last_statement.expression, NormalVariableExpression):
-        result_expression = result_statement_list.pop().expression
-
         if assign_result_to is not None:
+            result_expression = result_statement_list.pop().expression
             result_statement_list.append(
                 NormalVariableReassignmentStatement(
                     variable=assign_result_to,