Added dialogue state back