Adapt to the new behaviour of data.table