//itk include
#include "itkLightObject.h"
-#if ITK_VERSION_MAJOR >= 4
- #include "itkTransformToDisplacementFieldSource.h"
+#if (ITK_VERSION_MAJOR == 4) && (ITK_VERSION_MINOR < 6)
+# include "itkTransformToDisplacementFieldSource.h"
#else
- #include "itkTransformToDeformationFieldSource.h"
+# include "itkTransformToDisplacementFieldFilter.h"
#endif
#include "itkAffineTransform.h"