diff --git a/cnn_classification_service/cnn_classifier.py b/cnn_classification_service/cnn_classifier.py
index df8570b..9e4c194 100644
--- a/cnn_classification_service/cnn_classifier.py
+++ b/cnn_classification_service/cnn_classifier.py
@@ -39,9 +39,10 @@ class Classifier(object):
         librosa.display.specshow(librosa.power_to_db(spectogram, ref=numpy.max))
 
         target_dir = tempfile.mkdtemp()
+        wav_basename = os.path.basename(wav_filename)
 
         # Change extension to jpg... mert 110% biztos vagyok benne hogy a keras nem bírná beolvasni máshogy
-        file_name = os.path.join(target_dir, "unknown", f"{wav_filename[:-4]}.jpg")
+        file_name = os.path.join(target_dir, "unknown", f"{wav_basename[:-4]}.jpg")
 
         matplotlib.pyplot.savefig(file_name, dpi=400, bbox_inches='tight', pad_inches=0)
         matplotlib.pyplot.close()