diff --git a/src/tests/architect_tests/test_builders.py b/src/tests/architect_tests/test_builders.py
index 621932c13..1a881a080 100644
--- a/src/tests/architect_tests/test_builders.py
+++ b/src/tests/architect_tests/test_builders.py
@@ -1,4 +1,5 @@
 import datetime
+from unittest import TestCase, mock
 import pandas as pd
 import testing.postgresql