From cb6a905561eee0f4375a738685dbbc107ddb5b7e Mon Sep 17 00:00:00 2001
From: William Blum <william.blum@microsoft.com>
Date: Mon, 23 May 2022 14:16:02 -0700
Subject: [PATCH] Fix first part of #46

---
 cyberbattle/simulation/generate_network.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/cyberbattle/simulation/generate_network.py b/cyberbattle/simulation/generate_network.py
index 3f0e6bb..6b3fe25 100644
--- a/cyberbattle/simulation/generate_network.py
+++ b/cyberbattle/simulation/generate_network.py
@@ -161,10 +161,13 @@ def create_cached_credential(node: NodeID, port: PortName) -> CredentialID:
 
     def add_leak_neighbors_vulnerability(
             node_id: m.NodeID,
-            library: m.VulnerabilityLibrary = {}) -> m.VulnerabilityLibrary:
+            library: Optional[m.VulnerabilityLibrary] = None) -> m.VulnerabilityLibrary:
         """Create random vulnerabilities
         that reveals immediate traffic neighbors from a given node"""
 
+        if not library:
+            library = {}
+
         rdp_neighbors = traffic_targets(node_id, 'RDP')
 
         if len(rdp_neighbors) > 0: