diff --git a/displayer.py b/displayer.py
index e21fe1d6..8159a8ee 100644
--- a/displayer.py
+++ b/displayer.py
@@ -134,8 +134,8 @@ def place_on_screen(points_and_rgbs, space_width, space_height):
 
 def get_file_path(name, extension):
     file_path = os.path.join(MOVIE_DIR, name)
-    if not file_path.endswith(".mp4"):
-        file_path += ".mp4"
+    if not file_path.endswith(extension):
+        file_path += extension
     directory = os.path.split(file_path)[0]
     if not os.path.exists(directory):
         os.makedirs(directory)