1 files changed, 2 insertions, 2 deletions
diff --git a/src/util/node_visitor.h b/src/util/node_visitor.h
index 1adc0ff20..06a1a83e8 100644
--- a/src/util/node_visitor.h
+++ b/src/util/node_visitor.h
@@ -50,7 +50,7 @@ public:
   /**
    * Performs the traversal.
    */
-  static void run(Visitor& visitor, TNode node) {
+  static typename Visitor::return_type run(Visitor& visitor, TNode node) {
 
     // Notify of a start
     visitor.start(node);
@@ -86,7 +86,7 @@ public:
     }
 
     // Notify that we're done
-    visitor.done(node);
+    return visitor.done(node);
   }
 
 };