Remove unnecessary flatten! method call.