diff --git a/tests/test_table.py b/tests/test_table.py
index f80e0d1..b4818d7 100644
--- a/tests/test_table.py
+++ b/tests/test_table.py
@@ -1,5 +1,8 @@
 import unittest
-from table import Table, Cell, TableRenderer
+from src.table import (
+    Table, TableRenderer, Cell, ANSI_BOLD_BRIGHT_WHITE, ANSI_WHITE,
+    ANSI_RED, ANSI_GREEN, ANSI_CYAN, ANSI_PURPLE, ANSI_GREY
+)
 
 
 class TestCell(unittest.TestCase):