Fix up pipeline.py