core/transformations.mako: delaying return in the input transformation until the end of the snippet

This make the positioning of ${store_same} inside the snippet irrelevant.