diff --git a/gain_viz/app.py b/gain_viz/app.py
index 3272b21..7ff96a0 100644
--- a/gain_viz/app.py
+++ b/gain_viz/app.py
@@ -12,6 +12,7 @@ app = Flask(__name__)
 
 # Path to save the plot image
 #PLOT_PATH = os.path.join(os.getcwd(), "plot.png")
+os.makedirs("/opt/gain_viz", exist_ok=True)
 PLOT_PATH = "/opt/gain_viz/plot.png"
 
 # Global variables for gain values