diff --git a/src/super_gradients/common/deprecate.py b/src/super_gradients/common/deprecate.py
index 0f77138e47..eb0025d126 100644
--- a/src/super_gradients/common/deprecate.py
+++ b/src/super_gradients/common/deprecate.py
@@ -112,7 +112,7 @@ def override(self, **entries):
 
     def decorator(func):
         def wrapper(*args, **training_params):
-            if deprecated_tparam_name in training_params:
+            if training_params.get(deprecated_tparam_name):
                 import super_gradients
 
                 is_still_supported = parse_version(super_gradients.__version__) < parse_version(removed_from)