Bugfix for BatchCollator missing_args