diff --git a/src/run.py b/src/run.py
index d8486f5..d5eee5c 100644
--- a/src/run.py
+++ b/src/run.py
@@ -9,6 +9,8 @@ import json
 import os
 import plotly.express as px
 import plotly.graph_objects as po
+import plotly.io as pio
+import io
 import pandas as pd
 import psycopg
 import sqlalchemy