diff --git a/src/__init__.py b/src/__init__.py
index 637f57f..e69de29 100644
--- a/src/__init__.py
+++ b/src/__init__.py
@@ -1,12 +0,0 @@
-# __init__.py
-
-import logging
-from datetime import datetime
-
-# Configure logging
-logging.basicConfig(
-    format='%(asctime)s : %(message)s',
-    level = logging.INFO
-)
-
-logging.info("freq-split-enhance/separation package has been imported.")
\ No newline at end of file
diff --git a/src/separation/__init__.py b/src/separation/__init__.py
index e69de29..637f57f 100644
--- a/src/separation/__init__.py
+++ b/src/separation/__init__.py
@@ -0,0 +1,12 @@
+# __init__.py
+
+import logging
+from datetime import datetime
+
+# Configure logging
+logging.basicConfig(
+    format='%(asctime)s : %(message)s',
+    level = logging.INFO
+)
+
+logging.info("freq-split-enhance/separation package has been imported.")
\ No newline at end of file
diff --git a/tests/test_separation.py b/tests/test_separation.py
index a297292..5241124 100644
--- a/tests/test_separation.py
+++ b/tests/test_separation.py
@@ -20,9 +20,9 @@ def test_demucs_separation_with_preprocessing():
     waveform, samplerate = read_audio(input_file)
     waveform = trim_audio(waveform, samplerate)
 
-    # Resample to 41000Hz
-    if samplerate != 41000:
-        print("Resampling audio to 41Khz")
+    # Resample to 44100Hz
+    if samplerate != 44100:
+        print("Resampling audio to 44.1Khz")
         waveform, samplerate = resample(waveform, samplerate, 41000)
 
     # Create a temporary file to save the processed audio