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 10899d1f9..5e0955d76 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
@@ -1411,6 +1411,7 @@ class AxolotlInputConfig(
                     "FSDP w/ gradient_accumulation_steps > 1 is broken with transformers==4.46.2. "
                     "Please use a lower value or switch to an older version of transformers."
                 )
+        return data
 
 
 class AxolotlConfigWCapabilities(AxolotlInputConfig):