diff --git a/demo3.py b/demo3.py
index 6f35280..dafc1b1 100644
--- a/demo3.py
+++ b/demo3.py
@@ -75,7 +75,7 @@ for y, x in goodPix:
 image2 = image.copy()
 for y, x in badPix:
     coord = np.array([x,y])
-    image2[y,x] = model.getValueOfPoint(coord, 8)[0]
+    image2[y,x] = model.getValueOfPoint(coord, 8)
 
 print("temps d'infération des",str(len(badPix)),"pixels:",str(round(time()-t,3)),"s")