Fix NFVBENCH-2 Cleanup after error causes a AttributeError exception

[nfvbench.git] / nfvbench / service_chain.py

diff --git a/nfvbench/service_chain.py b/nfvbench/service_chain.py
index 104cfb4..0a5fd13 100644 (file)
--- a/nfvbench/service_chain.py
+++ b/nfvbench/service_chain.py
@@ -134,5 +134,7 @@ class ServiceChain(object):
         return self.stats_manager.get_version()
 
     def close(self):
-        self.stage_manager.close()
-        self.stats_manager.close()
+        if self.stage_manager:
+            self.stage_manager.close()
+        if self.stats_manager:
+            self.stats_manager.close()