diff --git a/nodes.py b/nodes.py
index 168c5b7..6ae9f92 100644
--- a/nodes.py
+++ b/nodes.py
@@ -166,7 +166,7 @@ class WarpFaceBack:
                                 for single_warp, single_mask in zip(warp, mask)]
                 full_mask = np.add.reduce(warped_masks, axis=0)[...,None]
                 swapped = np.add.reduce([
-                    cv2.warpAffine(single_crop.numpy(),
+                    cv2.warpAffine(single_crop.cpu().numpy(),
                                 cv2.invertAffineTransform(single_warp),
                                 image.shape[1::-1]
                                 ) * single_mask[..., None]