diff --git a/src/skrubview/_data/templates/dataframe-columns.css b/src/skrubview/_data/templates/dataframe-columns.css
index 39430ca..aad5465 100644
--- a/src/skrubview/_data/templates/dataframe-columns.css
+++ b/src/skrubview/_data/templates/dataframe-columns.css
@@ -1,4 +1,4 @@
-{% from "utils.j2" import compact_id, collapse_const_id %}
+{% from "utils.j2" import compact_id, collapse_const_id with context %}
 
 {{ report_id_selector }}.skrubview-report .skrubview-float-end {
     margin-inline-start: auto;
diff --git a/src/skrubview/_data/templates/dataframe-columns.html b/src/skrubview/_data/templates/dataframe-columns.html
index 25a6fae..0c02518 100644
--- a/src/skrubview/_data/templates/dataframe-columns.html
+++ b/src/skrubview/_data/templates/dataframe-columns.html
@@ -1,3 +1,4 @@
+{% from "utils.j2" import compact_id, collapse_const_id with context %}
 <div class="skrubview-wrapper">
     <details open="">
         <summary><h2>Columns</h2></summary>
diff --git a/src/skrubview/_data/templates/report.html b/src/skrubview/_data/templates/report.html
index 9828f9d..353b21f 100644
--- a/src/skrubview/_data/templates/report.html
+++ b/src/skrubview/_data/templates/report.html
@@ -1,5 +1,3 @@
-{% from "utils.j2" import compact_id, collapse_const_id, report_id_selector %}
-
 <div class="skrubview-report" id="{{ report_id }}">
     <div class="skrubview-dataframe-summary skrubview-wrapper skrubview-text">
         {% if summary.title %}