Small modification to the pipeline.