BUG: forgot to use affine registered image for 1-step reg