Fix state transition test-case that broke in d768330