Fix dataset test after attributes have been added