fix test

davidbailey · Dec 9, 2019 · e3718d3 · e3718d3
1 parent 90e8166
commit e3718d3
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/tests/test_vulnerability.py b/tests/test_vulnerability.py
@@ -3,7 +3,14 @@
 """
 import unittest
 
-from rail import Control, ThreatEvent, Tree, Vulnerability, Vulnerabilities
+from rail import (
+    Control,
+    ThreatEvent,
+    ThreatSources,
+    Tree,
+    Vulnerability,
+    Vulnerabilities,
+)
 
 
 class TestVulnerabilities(unittest.TestCase):