Rename stage arg to state