Implement expected behaviour from idAttribute prop