Forgot to actually set the state-check...