fix: Add state_type field if not exist