Fix test breaking to state selection