fixed dataset problem