diff --git a/cmd/dwi2tensor.cpp b/cmd/dwi2tensor.cpp
index c9ced7dc97..c814732552 100644
--- a/cmd/dwi2tensor.cpp
+++ b/cmd/dwi2tensor.cpp
@@ -220,7 +220,6 @@ void run ()
   Header header (dwi);
   header.datatype() = DataType::Float32;
   header.ndim() = 4;
-  DWI::stash_DW_scheme (header, grad);
   PhaseEncoding::clear_scheme (header);
 
   Image<value_type> predict;
@@ -228,6 +227,7 @@ void run ()
   if (opt.size())
     predict = Image<value_type>::create (opt[0][0], header);
 
+  DWI::stash_DW_scheme (header, grad);
   header.size(3) = 6;
   auto dt = Image<value_type>::create (argument[1], header);