Correcting dataset_controller_tests