Fixed small bug in pipeline