diff --git a/src/app_controller.py b/src/app_controller.py
index 0267ef5..9856cf6 100644
--- a/src/app_controller.py
+++ b/src/app_controller.py
@@ -1890,12 +1890,12 @@ class AppController:
   self._update_cached_stats()
 
  def _update_cached_stats(self) -> None:
-  import ai_client
+  from src import ai_client
   self._cached_cache_stats = ai_client.get_gemini_cache_stats()
   self._cached_tool_stats = dict(self._tool_stats)
 
  def clear_cache(self) -> None:
-  import ai_client
+  from src import ai_client
   ai_client.cleanup()
   self._update_cached_stats()