diff --git a/exp/kerokero/test.py b/exp/kerokero/test.py
--- a/exp/kerokero/test.py
+++ b/exp/kerokero/test.py
@@ -20,7 +20,7 @@ print("loading data...")
 print("done")
 
 for img in testImages:
-	label=model.predict(np.reshape(img,(1,224,224)))
+	label=model.predict(np.reshape(img,(1,224,224,1)))
 	print(label)
 	points=[]
 	for i in range(4):