1 files changed, 2 insertions, 3 deletions
diff --git a/examples/vpn.py b/examples/vpn.py
index 4ad9f6d..b2f3c81 100755
--- a/examples/vpn.py
+++ b/examples/vpn.py
@@ -3,6 +3,7 @@
 # An example script using the rumba package
 
 from rumba.model import *
+from rumba.utils import ExperimentManager
 
 # import testbed plugins
 import rumba.testbeds.emulab as emulab
@@ -49,8 +50,6 @@ exp = our.Experiment(tb, nodes = [a, b, c, d])
 
 print(exp)
 
-#try:
+# with ExperimentManager(exp):
 #    exp.swap_in()
 #    exp.bootstrap_prototype()
-#finally:
-#    exp.swap_out()