fix imputs again

Author: josephmje

dmriprep/workflows/dwi/util.py

@@ -90,7 +90,7 @@ def init_dwi_reference_wf(omp_nthreads, dwi_file=None,
 
     extract_b0 = pe.Node(ExtractB0(), name='extract_b0')
 
-    reg_b0 = fsl.MCFLIRT(ref_vol=0, interpolation='sinc')
+    reg_b0 = pe.Node(fsl.MCFLIRT(ref_vol=0, interpolation='sinc'), name='reg_b0')
 
     pre_mask = pe.Node(afni.Automask(dilate=1, outputtype='NIFTI_GZ'),
                        name='pre_mask')
@@ -107,9 +107,9 @@ def init_dwi_reference_wf(omp_nthreads, dwi_file=None,
         (extract_b0, reg_b0, [('out_file', 'in_file')]),
         (reg_b0, pre_mask, [('out_file', 'in_file')]),
         (reg_b0, rescale_b0, [('out_file', 'in_file')]),
-        (pre_mask, rescale_b0, [('out_file', 'pre_mask')]),
+        (pre_mask, rescale_b0, [('out_file', 'mask_file')]),
         (rescale_b0, enhance_and_skullstrip_dwi_wf, [('out_file', 'inputnode.in_file')]),
-        (pre_mask, enhance_and_skullstrip_dwi_wf, [('out_file', 'inputnode.mask_file')]),
+        (pre_mask, enhance_and_skullstrip_dwi_wf, [('out_file', 'inputnode.pre_mask')]),
         (validate, outputnode, [('out_file', 'dwi_file'),
                                 ('out_report', 'validation_report')]),
         (rescale_b0, outputnode, [('out_file', 'raw_ref_image')]),