Fixed some variables in the state class

Signed-off-by: Ian bell <ian.h.bell@gmail.com>