diff --git a/tsfm_public/toolkit/get_model.py b/tsfm_public/toolkit/get_model.py
index ace81075..bfb9a71f 100644
--- a/tsfm_public/toolkit/get_model.py
+++ b/tsfm_public/toolkit/get_model.py
@@ -148,8 +148,8 @@ def get_model(
                 else:
                     freq_prefix = None
             else:
-                raise Exception(
-                    "In current implementation, set freq_prefix_tuning to None for automatic model selection accordingly.."
+                raise ValueError(
+                    "In the current implementation, set `freq_prefix_tuning` to None for automatic model selection accordingly."
                 )
                 if freq_prefix_tuning:
                     freq_prefix = "freq"