diff --git a/gemma/gm/ckpts/_checkpoint.py b/gemma/gm/ckpts/_checkpoint.py
index 60443184..68c7e9b3 100644
--- a/gemma/gm/ckpts/_checkpoint.py
+++ b/gemma/gm/ckpts/_checkpoint.py
@@ -268,7 +268,7 @@ def load_params(
       raise ValueError(
           'The input params provided to `load_params()` should be the raw'
           " model params matching the Flax `model.init()['params']` structure."
-          f' Got: {_CheckpointType.NESTED}'
+          f' Got: {params.type}'
       )
     if text_only and params.has_mm_params:
       raise ValueError(