Adding support for device assignment state.