diff --git a/piel/tools/amaranth/construct.py b/piel/tools/amaranth/construct.py
index c5250e24..5e730e47 100644
--- a/piel/tools/amaranth/construct.py
+++ b/piel/tools/amaranth/construct.py
@@ -8,8 +8,7 @@
 - "memory"
 """
 
-import amaranth as am
-from ...types.digital import TruthTable, LogicImplementationType
+from piel.types.digital import TruthTable, LogicImplementationType
 
 
 def construct_amaranth_module_from_truth_table(
@@ -43,6 +42,7 @@ def construct_amaranth_module_from_truth_table(
         >>> )
         >>> am_module = construct_amaranth_module_from_truth_table(my_truth_table)
     """
+    import amaranth as am
 
     # Extract inputs and outputs from the truth table
     inputs = truth_table.input_ports
diff --git a/tests/import/test_import_core.py b/tests/import/test_import_core.py
index e69de29b..e2c977ac 100644
--- a/tests/import/test_import_core.py
+++ b/tests/import/test_import_core.py
@@ -0,0 +1 @@
+import piel
diff --git a/tests/import/test_import_experimental.py b/tests/import/test_import_experimental.py
index e69de29b..4fcf9a9c 100644
--- a/tests/import/test_import_experimental.py
+++ b/tests/import/test_import_experimental.py
@@ -0,0 +1 @@
+import piel.experimental