Fixed missing constant

STATES used to be a global constant but it is not
anymore. Shifting it accordingly.