diff --git a/google/genai/tunings.py b/google/genai/tunings.py
index 9dcf43e8..53ca9dcb 100644
--- a/google/genai/tunings.py
+++ b/google/genai/tunings.py
@@ -1503,7 +1503,7 @@ def _display_link(
         </script>
     """
 
-    from IPython.core.display import display
+    from IPython.display import display
     from IPython.display import HTML
 
     display(HTML(html))