fix dataset bug