diff --git a/firebaseai/testapp/Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs b/firebaseai/testapp/Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs
index edcbdee6..d4b4a162 100644
--- a/firebaseai/testapp/Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs
+++ b/firebaseai/testapp/Assets/Firebase/Sample/FirebaseAI/UIHandlerAutomated.cs
@@ -75,7 +75,10 @@ protected override void Start()
         TestCountTokens,
         TestYoutubeLink,
         TestGenerateImage,
+#if !(FIREBASE_RUNNING_FROM_CI && !UNITY_EDITOR)
+        // Disabled from CI, because of rate limit issues
         TestImagenGenerateImage,
+#endif
         TestImagenGenerateImageOptions,
         TestThinkingBudget,
         TestIncludeThoughts,
@@ -83,7 +86,10 @@ protected override void Start()
         TestUrlContext,
         TestTemplateGenerateContent,
         TestTemplateGenerateContentStream,
+#if !(FIREBASE_RUNNING_FROM_CI && !UNITY_EDITOR)
+        // Disabled from CI, because of rate limit issues
         TestTemplateImagenGenerateImage,
+#endif
       };
       // Set of tests that only run the single time.
       Func<Task>[] singleTests = {