Made small fix to lost state