diff --git a/src/wireviz/wv_bom.py b/src/wireviz/wv_bom.py
index 56df752..c60ea0d 100644
--- a/src/wireviz/wv_bom.py
+++ b/src/wireviz/wv_bom.py
@@ -4,8 +4,8 @@ from dataclasses import asdict
 from itertools import groupby
 from typing import Any, Dict, List, Optional, Tuple, Union
 
-from wireviz.DataClasses import AdditionalComponent, Cable, Color, Connector
-from wireviz.wv_colors import translate_color
+from wireviz.DataClasses import AdditionalComponent, Cable, Connector
+from wireviz.wv_colors import Color, translate_color
 from wireviz.wv_gv_html import html_bgcolor_attr, html_line_breaks
 from wireviz.wv_helper import clean_whitespace
 
diff --git a/src/wireviz/wv_gv_html.py b/src/wireviz/wv_gv_html.py
index 0b843db..065dd1b 100644
--- a/src/wireviz/wv_gv_html.py
+++ b/src/wireviz/wv_gv_html.py
@@ -3,8 +3,7 @@
 from typing import List, Optional, Union
 import re
 
-from wireviz.DataClasses import Color
-from wireviz.wv_colors import translate_color
+from wireviz.wv_colors import Color, translate_color
 from wireviz.wv_helper import remove_links
 
 def nested_html_table(rows: List[Union[str, List[Optional[str]], None]], table_attrs: str = '') -> str: