diff --git a/cavemanml.py b/cavemanml.py
index 9e1a7d9..b1bfa5c 100644
--- a/cavemanml.py
+++ b/cavemanml.py
@@ -66,12 +66,12 @@ def main():
         model.eval()
         enhance_audio(
             model,
-            "./mirror_mirror_compressed_64.mp3",
-            "./mirror_mirror_decompressed_64_mse.wav",
+            "./examples/mirror_mirror/mirror_mirror_compressed_64.mp3",
+            "./examples/mirror_mirror/mirror_mirror_decompressed_64_mse.wav",
         )
 
         # Load
-        x, sr = librosa.load("./mirror_mirror_compressed_64.mp3", sr=SR)
+        x, sr = librosa.load("./examples/mirror_mirror/mirror_mirror_compressed_64.mp3", sr=SR)
 
         # Convert to log-mag
         X = audio_to_logmag(x)  # (513, T)
@@ -90,7 +90,7 @@ def main():
         import soundfile as sf
 
         # Save
-        sf.write("./mirror_mirror_compressed_64_STFT.mp3", y_reconstructed, sr)
+        sf.write("./examples/mirror_mirror/mirror_mirror_compressed_64_STFT.mp3", y_reconstructed, sr)
         return
 
     # Data