Don't try to access a nonexistent return expression

diff --git a/source/programcompiler.cpp b/source/programcompiler.cpp
index 1a3cd34fae5067201f81dabf8ffcdf76121b0d59..c2ebb4b96ad3e474b2f636578e404601282b82ad 100644 (file)
--- a/source/programcompiler.cpp
+++ b/source/programcompiler.cpp
@@ -557,8 +557,12 @@ void ProgramCompiler::Formatter::visit(Iteration &iter)
 
 void ProgramCompiler::Formatter::visit(Return &ret)
 {
 
 void ProgramCompiler::Formatter::visit(Return &ret)
 {

-       formatted += "return ";
-       ret.expression->visit(*this);
+       formatted += "return";
+       if(ret.expression)
+       {
+               formatted += ' ';
+               ret.expression->visit(*this);
+       }

        formatted += ';';
 }
 
        formatted += ';';
 }