Correct else position

[clitk.git] / registration / clitkMatrixTransformToVFGenericFilter.h

diff --git a/registration/clitkMatrixTransformToVFGenericFilter.h b/registration/clitkMatrixTransformToVFGenericFilter.h
index 717e938a4b667b86dfaebc3eea570d578789e222..cc6ae0016dd173533959b575e37fe988622f84ab 100644 (file)
--- a/registration/clitkMatrixTransformToVFGenericFilter.h
+++ b/registration/clitkMatrixTransformToVFGenericFilter.h
@@ -37,9 +37,13 @@
 //itk include
 #include "itkLightObject.h"
 #if ITK_VERSION_MAJOR >= 4
-  #include "itkTransformToDisplacementFieldSource.h"
+#  if ITK_VERSION_MINOR < 6
+#    include "itkTransformToDisplacementFieldSource.h"
+#  else
+#    include "itkTransformToDisplacementFieldFilter.h"
+#  endif
 #else
-  #include "itkTransformToDeformationFieldSource.h"
+#  include "itkTransformToDeformationFieldSource.h"
 #endif
 #include "itkAffineTransform.h"