fix: Hopefully last fix for python 2.7 & reportlab.

MISP · Feb 28, 2019 · c35a2ae · c35a2ae
1 parent 0d6aa42
commit c35a2ae
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_reportlab.py b/tests/test_reportlab.py
@@ -4,7 +4,6 @@
 import unittest
 from pymisp import MISPEvent
 
-from pymisp.tools import reportlab_generator
 
 import sys
 import os
@@ -15,6 +14,8 @@
 if sys.version_info < (3, 6):
     print('This test suite requires Python 3.6+, breaking.')
     sys.exit(0)
+else:
+    from pymisp import reportlab_generator
 
 
 class TestMISPEvent(unittest.TestCase):