Adding conditional to Stage header