input_lengths for SelectedStep works better