diff --git a/registration/model.py b/registration/model.py
index <HASH>..<HASH> 100644
--- a/registration/model.py
+++ b/registration/model.py
@@ -42,10 +42,10 @@ class RegistrationModel(object):
             (k, v) = item
             return self.transformations[k].apply(v)
 
-        return images.map(apply, with_keys=True)
+        return images.map(apply, value_shape=images.value_shape, dtype=images.dtype, with_keys=True)
 
     def __repr__(self):
         s = self.__class__.__name__
         s += '\nlength: %g' % len(self.transformations)
         s += '\nalgorithm: ' + self.algorithm
-        return s
\ No newline at end of file
+        return s
skip shape and dtype check during transform