6 files changed, 7 insertions, 6 deletions
diff --git a/test/__init__.py b/test/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/test/__init__.py
diff --git a/test/dfatool b/test/dfatool
new file mode 120000
index 0000000..dc598c5
--- /dev/null
+++ b/test/dfatool
@@ -0,0 +1 @@
+../lib
+\ No newline at end of file
diff --git a/test/test_codegen.py b/test/test_codegen.py
index fb0d387..981117b 100755
--- a/test/test_codegen.py
+++ b/test/test_codegen.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 
-from automata import PTA
-from codegen import get_simulated_accountingmethod
+from dfatool.automata import PTA
+from dfatool.codegen import get_simulated_accountingmethod
 import unittest
 
 example_json_1 = {
diff --git a/test/test_pta.py b/test/test_pta.py
index cf8bce4..9f0778d 100755
--- a/test/test_pta.py
+++ b/test/test_pta.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 
-from automata import PTA
+from dfatool.automata import PTA
 import unittest
 import yaml
 
diff --git a/test/test_ptamodel.py b/test/test_ptamodel.py
index 6bd2f6a..7d501e6 100755
--- a/test/test_ptamodel.py
+++ b/test/test_ptamodel.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 
-from dfatool import PTAModel, RawData, pta_trace_to_aggregate
+from dfatool.dfatool import PTAModel, RawData, pta_trace_to_aggregate
 import os
 import unittest
 import pytest
diff --git a/test/test_timingharness.py b/test/test_timingharness.py
index 8d5ec20..c8a422c 100755
--- a/test/test_timingharness.py
+++ b/test/test_timingharness.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 
-from dfatool import AnalyticModel, TimingData, pta_trace_to_aggregate
-from parameters import prune_dependent_parameters
+from dfatool.dfatool import AnalyticModel, TimingData, pta_trace_to_aggregate
+from dfatool.parameters import prune_dependent_parameters
 import unittest