diff --git a/torchrec/distributed/train_pipeline/utils.py b/torchrec/distributed/train_pipeline/utils.py
index 76cf87370..791f73703 100644
--- a/torchrec/distributed/train_pipeline/utils.py
+++ b/torchrec/distributed/train_pipeline/utils.py
@@ -229,8 +229,6 @@ def _build_args_kwargs(
                 kwargs[arg_info.name] = arg
             else:
                 args.append(arg)
-        else:
-            args.append(None)
     return args, kwargs