When collecting shapes from the outputs of the inner function, we should
keep track that mit_mot arguments have multiple outputs for one argument.