X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=examples%2Fadd_model.py;h=3e46490b20f6f45fcb89a173407c848467b03c74;hb=eac6d53d640fec2d1adab425c0ce891136b5784b;hp=e70345038d97a7d684d02b2342d7c6afab85b424;hpb=98e277e099c01117cfb694e25746d8900649b3e3;p=osm%2FN2VC.git

diff --git a/examples/add_model.py b/examples/add_model.py
index e703450..3e46490 100644
--- a/examples/add_model.py
+++ b/examples/add_model.py
@@ -51,14 +51,16 @@ async def main():
         print("Destroying model")
         await controller.destroy_model(model.info.uuid)
 
-    except Exception as e:
+    except Exception:
         LOG.exception(
             "Test failed! Model {} may not be cleaned up".format(model_name))
 
     finally:
         print('Disconnecting from controller')
-        await model.disconnect()
+        if model:
+            await model.disconnect()
         await controller.disconnect()
 
 
-loop.run(main())
+if __name__ == '__main__':
+    loop.run(main())