diff --git a/test_iris.py b/test_iris.py
index 7cf6bd4..c09ba80 100644
--- a/test_iris.py
+++ b/test_iris.py
@@ -61,9 +61,9 @@ def centerCropSquare(img, center, side=None, scaleWRTHeight=None):
 
 
 plt.imshow(img, zorder=1)
-x, y = eye[:, 0], eye[:, 1]
+x, y = eye[0,:, 0], eye[0,:, 1]
 plt.scatter(x, y, zorder=2, s=1.0)
-x, y = iris[:, 0], iris[:, 1]
+x, y = iris[0,:, 0], iris[0,:, 1]
 plt.scatter(x, y, zorder=2, s=1.0, c='r')
 plt.show()