I have added a line into the pipeline.py.