Add step state test