Add DWI dataset back