st/vega: fix up vega state tracker to use cso changes