fix getunitstate handling