Make sure we load current state on restarts