- #include "itkTransformToDisplacementFieldSource.h"
-#else
- #include "itkTransformToDeformationFieldSource.h"
+# if ITK_VERSION_MINOR < 6
+# include "itkTransformToDisplacementFieldSource.h"
+# else
+# include "itkTransformToDisplacementFieldFilter.h"
+# endif