diff --git a/src/ntp_routes.py b/src/ntp_routes.py
index 15114e2..3093988 100644
--- a/src/ntp_routes.py
+++ b/src/ntp_routes.py
@@ -1,3 +1,4 @@
+from flask import Flask, session, g, render_template_string
 from loguru import logger
 import json
 import plotly.express as px
diff --git a/src/pv_routes.py b/src/pv_routes.py
index 191e793..96bd77c 100644
--- a/src/pv_routes.py
+++ b/src/pv_routes.py
@@ -1,3 +1,4 @@
+from flask import Flask, session, g, render_template_string
 from loguru import logger
 import json
 import plotly.express as px