diff --git a/test/test_audio.py b/test/test_audio.py
index 373fde3..9c39464 100644
--- a/test/test_audio.py
+++ b/test/test_audio.py
@@ -1,5 +1,5 @@
 import pytest
-from ..scraibe.audio import AudioProcessor
+from scraibe.audio import AudioProcessor
 import torch
 
 
diff --git a/test/test_diarisation.py b/test/test_diarisation.py
index 9c287c1..1976016 100644
--- a/test/test_diarisation.py
+++ b/test/test_diarisation.py
@@ -1,7 +1,7 @@
 import pytest
 import os
 from unittest import mock
-from ..scraibe import Diariser
+from scraibe import Diariser
 
 
 
diff --git a/test/test_transcriber.py b/test/test_transcriber.py
index 239f3f0..ef17951 100644
--- a/test/test_transcriber.py
+++ b/test/test_transcriber.py
@@ -1,6 +1,6 @@
 import pytest
 from unittest.mock import patch
-from ..scraibe import Transcriber
+from scraibe import Transcriber