Fix issue #98

I tested it and it seems to work. But I don't understand completely how the AST in FunScript works, so please have a look to be sure I am not breaking anything. Thanks in advance!