Fix for dataset problem.