fix remove dataset bug