diff --git a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py
index 6eaf34c547..2264c38cd6 100644
--- a/src/axolotl/utils/config/models/input/v0_4_1/__init__.py
+++ b/src/axolotl/utils/config/models/input/v0_4_1/__init__.py
@@ -689,10 +689,10 @@ def check_saves(cls, data):
     @classmethod
     def check_push_save(cls, data):
         if data.get("hub_model_id") and not (
-            data.get("save_steps") or data.get("saves_per_epoch")
+            data.get("save_strategy")
         ):
             LOG.warning(
-                "hub_model_id is set without any models being saved. To save a model, set either save_steps or saves_per_epoch."
+                "hub_model_id is set without any models being saved. To save a model, set save_strategy."
             )
         return data