diff --git a/src/separation/demucs_wrapper.py b/src/separation/demucs_wrapper.py
index 0b7fc5b..00958b2 100644
--- a/src/separation/demucs_wrapper.py
+++ b/src/separation/demucs_wrapper.py
@@ -4,7 +4,7 @@ import soundfile as sf
 import tempfile
 from pathlib import Path
 
-def separate_audio_with_demucs(input_file, sample_rate, output_dir: str):
+def separate_audio_with_demucs(input_file, output_dir: str):
     """
     Use subprocess to run Demucs separation on an input audio file.
     Parameters:
diff --git a/tests/test_separation.py b/tests/test_separation.py
index 5241124..c98ddf6 100644
--- a/tests/test_separation.py
+++ b/tests/test_separation.py
@@ -36,7 +36,7 @@ def test_demucs_separation_with_preprocessing():
     new_audio_path = os.path.join(dir_path, f"{file_name}.wav")
     os.rename(temp_audio_path, new_audio_path)
 
-    separate_audio_with_demucs(new_audio_path, samplerate, output_path)
+    separate_audio_with_demucs(new_audio_path, output_path)
 
     # Verify the htdemucs folder exists
     demucs_dir = Path(output_path) / 'htdemucs'