diff --git a/apps/wizard/app_pages/datainsight_robot.py b/apps/wizard/app_pages/datainsight_robot.py
index 0ff19cc29e3..3ba9f0c0dd4 100644
--- a/apps/wizard/app_pages/datainsight_robot.py
+++ b/apps/wizard/app_pages/datainsight_robot.py
@@ -1,17 +1,11 @@
-import base64
-from pathlib import Path
 from typing import cast
-from urllib import parse
 
-import pandas as pd
-import requests
 import streamlit as st
 from owid.catalog.charts import Chart
 
 from apps.utils.gpt import OpenAIWrapper
 from apps.wizard.app_pages.insights import (
     fetch_chart_data,
-    fetch_data,
     get_grapher_thumbnail,
     get_thumbnail_url,
     list_charts,
@@ -169,7 +163,7 @@ def get_trajectory_prompt(base_prompt: str, slug: str) -> str:
     all_charts = list_charts(conn)
     slugs = st.multiselect(
         label="Grapher slug",
-        options=[None] + all_charts,
+        options=all_charts,
         help="Introduce the URL to a Grapher URL. Query parameters work!",
         key="tab2_url",
     )
diff --git a/apps/wizard/app_pages/insights.py b/apps/wizard/app_pages/insights.py
index 26745df1579..acdb723b1b7 100644
--- a/apps/wizard/app_pages/insights.py
+++ b/apps/wizard/app_pages/insights.py
@@ -1,14 +1,14 @@
 import base64
 import datetime as dt
 import json
+from pathlib import Path
 from typing import Literal
+from urllib import parse
 
 import pandas as pd
 import requests
 from dateutil.parser import parse as date_parse
 
-from etl.db import get_connection
-
 
 def get_thumbnail_url(grapher_url: str) -> str:
     """