diff --git a/app/routes.py b/app/routes.py
index 884dd06..47e1942 100644
--- a/app/routes.py
+++ b/app/routes.py
@@ -2,6 +2,7 @@ import argparse
 import base64
 import io
 import json
+import os
 import pickle
 import urllib.parse as urlparse
 import uuid