Minor edit to final stage pipeline