--- a/dmri.py	Sun Nov 05 07:43:52 2017 -0500
+++ b/dmri.py	Sun Nov 05 08:34:54 2017 -0500
@@ -18,12 +18,12 @@
 
 # Get input data.
 input_dir = args.drmi_dataset
-if args.drmi_dataset == 'sherbrooke_3shell':
+if input_dir == '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':
+elif input_dir == 'stanford_hardi':
     fetch_stanford_hardi()
     fdwi = os.path.join(input_dir, 'HARDI150.nii.gz')
     fbval = os.path.join(input_dir, 'HARDI150.bval')