Use proper data attribute