fix data loading issue for variable selection