diff --git a/examples/gemma/sampler_test.py b/examples/gemma/sampler_test.py
index f2c36720b..911a18247 100644
--- a/examples/gemma/sampler_test.py
+++ b/examples/gemma/sampler_test.py
@@ -553,7 +553,7 @@ def test_models_from_kaggle(self, url):
       import kagglehub
 
       has_kagglehub_dep = True
-    except ModuleNotFoundError:
+    except (ImportError, ModuleNotFoundError):
       has_kagglehub_dep = False
 
     if not (has_kaggle_creds and has_kagglehub_dep):