fix extracting loop conditions for 'while' and 'until' loops

1 files changed, 3 insertions, 2 deletions

diff --git a/TurtleArt/taprimitive.py b/TurtleArt/taprimitive.py
index 6b0c515..db4e0f9 100644
--- a/TurtleArt/taprimitive.py
+++ b/TurtleArt/taprimitive.py
@@ -327,10 +327,11 @@ class Primitive(object):
                 if controller == Primitive.controller_forever:
                     condition_ast = ast.Name(id="True", ctx=ast.Load)
                 elif controller == Primitive.controller_while:
-                    condition_ast = new_arg_asts[0].args[0]
+                    condition_ast = new_arg_asts[0].args[0].body
                 elif controller == Primitive.controller_until:
+                    pos_cond_ast = new_arg_asts[0].args[0].body
                     condition_ast = ast.UnaryOp(op=ast.Not,
-                                              operand=new_arg_asts[0].args[0])
+                                                operand=pos_cond_ast)
                 else:
                     raise ValueError("unknown loop controller: " +
                                      repr(controller))