diff --git a/visJS2jupyter/visJS_module.py b/visJS2jupyter/visJS_module.py
index 90359be..1c02b3d 100644
--- a/visJS2jupyter/visJS_module.py
+++ b/visJS2jupyter/visJS_module.py
@@ -415,7 +415,7 @@ def visjs_network(nodes_dict, edges_dict,
       script = """
         {}
         function setUpFrame() {{
-          window.runVis({}, {});
+          return window.runVis({}, {});
         }}{}
       """.format(result["script"], dumps(nodes_dict), dumps(edges_dict), (("\n" + result["run"]) if output == "zeppelin" else ""))
       head = """
@@ -1203,6 +1203,7 @@ def stringify_bool(var):
 
 
        console.log( "ready!" );
+       return myNetwork;
     }
     """