From 69be4328ce0a90a25758a7d158b8bc4ee973ce29 Mon Sep 17 00:00:00 2001
From: Grant Sanderson <grantsanderson7@gmail.com>
Date: Wed, 6 Jan 2016 17:51:37 -0800
Subject: [PATCH] Small bug in how displayer had been refactored

---
 displayer.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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)