diff --git a/enhancer/utils/io.py b/enhancer/utils/io.py
index 510ab86..04015e6 100644
--- a/enhancer/utils/io.py
+++ b/enhancer/utils/io.py
@@ -29,6 +29,8 @@ class Audio:
             if os.path.exists(audio):
                 audio,sampling_rate = librosa.load(audio,sr=sampling_rate,mono=False,
                 offset=offset,duration=duration)
+                if len(audio.shape) == 1:
+                    audio = audio.reshape(1,-1)
             else:
                 raise FileNotFoundError(f"File {audio} deos not exist")
         elif isinstance(audio,np.ndarray):