fix pipeline stage count