add annotation for State module