Add a correct simplify method for StreamTransform