diff --git a/Dockerfile b/Dockerfile
index e26e6a4..4a29e0f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,6 +1,6 @@
 FROM ubuntu:22.04
 LABEL authors="calvinchai"
-
+# TODO: Optimize this
 WORKDIR /app
 COPY ./requirements.txt /app/requirements.txt
 RUN apt update
diff --git a/scripts/setup.py b/scripts/setup.py
index 88856c4..98b839e 100644
--- a/scripts/setup.py
+++ b/scripts/setup.py
@@ -11,7 +11,7 @@
 
 
 def check_directory():
-    optional_dirs = ['logs', 'workdir', 'settings', 'modules', 'plugins']
+    optional_dirs = ['logs', 'workdir', 'settings', 'modules', 'plugins', 'output']
 
     for d in optional_dirs:
         if not app_path.joinpath(d).exists():