--- a/dmri.py	Sun Nov 05 07:35:51 2017 -0500
+++ b/dmri.py	Sun Nov 05 07:43:52 2017 -0500
@@ -20,11 +20,14 @@
 input_dir = args.drmi_dataset
 if args.drmi_dataset == 'sherbrooke_3shell':
     fetch_sherbrooke_3shell()
+    fdwi = os.path.join(input_dir, 'HARDI193.nii.gz')
+    fbval = os.path.join(input_dir, 'HARDI193.bval')
+    fbvec = os.path.join(input_dir, 'HARDI193.bvec')
 elif args.drmi_dataset == 'stanford_hardi':
     fetch_stanford_hardi()
-fdwi = os.path.join(input_dir, 'HARDI193.nii.gz')
-fbval = os.path.join(input_dir, 'HARDI193.bval')
-fbvec = os.path.join(input_dir, 'HARDI193.bvec')
+    fdwi = os.path.join(input_dir, 'HARDI150.nii.gz')
+    fbval = os.path.join(input_dir, 'HARDI150.bval')
+    fbvec = os.path.join(input_dir, 'HARDI150.bvec')
 # Load the dMRI datasets.
 img = nibabel.load(fdwi)
 data = img.get_data()