Small optimization in source transformer