diff --git a/.vscode/settings.json b/.vscode/settings.json
index de288e1eab6a537caed290ce0fc5e70be7ae9dba..d35c5419942948e9f9dccac24c74e65b78fc296f 100644
--- a/.vscode/settings.json
+++ b/.vscode/settings.json
@@ -1,3 +1,12 @@
 {
-    "python.formatting.provider": "black"
+    "python.formatting.provider": "black",
+    "python.testing.unittestArgs": [
+        "-v",
+        "-s",
+        "./tests",
+        "-p",
+        "test*.py"
+    ],
+    "python.testing.pytestEnabled": false,
+    "python.testing.unittestEnabled": true
 }
\ No newline at end of file
diff --git a/PyNutil/processing/counting_and_load.py b/PyNutil/processing/counting_and_load.py
index 4fdcf3204e32c6df394dcf378d17f14d2d45f268..f7498e2640ab255965e8052b2017e2f463f49197 100644
--- a/PyNutil/processing/counting_and_load.py
+++ b/PyNutil/processing/counting_and_load.py
@@ -209,7 +209,10 @@ def warp_image(image, triangulation, rescaleXY):
     newY[newY >= reg_h] = reg_h - 1
     newX[newX < 0] = 0
     newY[newY < 0] = 0
-    new_image = image[newY, newX]
+    new_image = np.zeros_like(image)
+    mask = (newX <= reg_w) & (newY <= reg_h) & (newX > 0) & (newY > 0)
+    new_image[mask] = image[mask]
+    new_image[~mask] = 0
     return new_image