Fix the dataset loading when in the code directory.