Use correct data attribute