fix data_dir unittest