diff --git a/music_processor.py b/music_processor.py
index 44bd4cd2f63c3de5fbfe20da4400b92686b52523..580443b4e7fb900277eaae4e228cdebdc56486f3 100644
--- a/music_processor.py
+++ b/music_processor.py
@@ -267,7 +267,7 @@ def music_for_train(serv, verbose=False, nhop=512, nffts=[1024, 2048, 4096], mel
         if verbose:
             print(songplace)
         
-        song = Audio(glob(songplace+"/*.wav")[0])
+        song = Audio(glob(songplace+"/*.ogg")[0])
         song.import_ass(glob(songplace+"/*.ass")[-1])
         song.data = (song.data[:, 0]+song.data[:, 1])/2
         songs.append(song)
@@ -280,7 +280,7 @@ def music_for_train(serv, verbose=False, nhop=512, nffts=[1024, 2048, 4096], mel
 
 def music_for_test(serv, deletemusic=True, verbose=False):
 
-    song = Audio(glob(serv+"/*.wav")[0], stereo=False)
+    song = Audio(glob(serv+"/*.ogg")[0], stereo=False)
     # song.import_tja(glob(serv+"/*.tja")[-1])
     song.feats = fft_and_melscale(song, include_zero_cross=False)
     with open('./data/pickles/test_data.pickle', mode='wb') as f: