diff --git a/pyproject.toml b/pyproject.toml
index 3b48471..bb66e2c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,7 +13,7 @@ dependencies = [
   'scikit-image == 0.19.3'
 ]
 
-[tool.hatch.build.targets]
+[tool.hatch.build.targets.wheel]
 packages = ["src/kafkajobs"]
 
 description = "A job queue based on Apache Kafka"
diff --git a/tests/test_jobs.py b/tests/test_jobs.py
index a808fcb..1613ed2 100644
--- a/tests/test_jobs.py
+++ b/tests/test_jobs.py
@@ -1,6 +1,6 @@
 import unittest
 
-from src.kafkajobs.jobqueue.queue import JobQueueProducer, JobQueueWorker
+from kafkajobs.jobqueue.queue import JobQueueProducer, JobQueueWorker
 
 class TestSum(unittest.TestCase):
     def test_pub_sub(self):