fixed dataset bug