BUG: Update the code for ITKv4.