diff --git a/test/test_modules.py b/test/test_modules.py
index c36c020..5218eaf 100644
--- a/test/test_modules.py
+++ b/test/test_modules.py
@@ -10,6 +10,7 @@ class TestOntologyToModule(unittest.TestCase):
 
     def setUp(self):
         SBOLFactory.clear()
+        self.maxDiff = None
 
     def tearDown(self):
         SBOLFactory.clear()
@@ -38,9 +39,9 @@ def test_figure_generation(self):
         figure_maker.generate(tmp)
         dot_source_actual = ''
         with open(os.path.join(tmp, 'Activity_abstraction_hierarchy')) as dot_file:
-            dot_source_actual = dot_file.read()
+            dot_source_actual = dot_file.read().replace(' ', '')
         with open(os.path.join(path, 'Activity_abstraction_hierarchy.dot')) as dot_file:
-            dot_source_expected = dot_file.read()
+            dot_source_expected = dot_file.read().replace(' ', '')
         self.assertEqual(dot_source_actual, dot_source_expected) 
 
     def test_figure_generation2(self):
@@ -54,9 +55,9 @@ def test_figure_generation2(self):
         figure_maker.generate(tmp)
         dot_source_actual = ''
         with open(os.path.join(tmp, 'Base_abstraction_hierarchy')) as dot_file:
-            dot_source_actual = dot_file.read()
+            dot_source_actual = dot_file.read().replace(' ', '')
         with open(os.path.join(path, 'Base_abstraction_hierarchy.dot')) as dot_file:
-            dot_source_expected = dot_file.read()
+            dot_source_expected = dot_file.read().replace(' ', '')
         self.assertEqual(dot_source_actual, dot_source_expected)