diff --git a/python/ngen_cal/tests/test_ngen.py b/python/ngen_cal/tests/test_ngen.py
index ad7f826d..e83c68ff 100644
--- a/python/ngen_cal/tests/test_ngen.py
+++ b/python/ngen_cal/tests/test_ngen.py
@@ -1,5 +1,15 @@
+import pathlib
+
 import pytest
-from ngen.cal.ngen import NgenExplicit, NgenIndependent, NgenUniform, NgenStrategy
+import pydantic
+from ngen.cal.ngen import (
+    Ngen,
+    NgenBase,
+    NgenExplicit,
+    NgenIndependent,
+    NgenStrategy,
+    NgenUniform,
+)
 
 
 def test_NgenExplicit_strategy_default_value():