Fix state bug

Signed-off-by: Tim Smart <tim@fostle.com>