diff --git a/exp/kerokero/test.py b/exp/kerokero/test.py
--- a/exp/kerokero/test.py
+++ b/exp/kerokero/test.py
@@ -10,13 +10,17 @@ from analyzer.corners import Corners
 
 parser=argparse.ArgumentParser()
 parser.add_argument("model")
-parser.add_argument("data_dir")
+parser.add_argument("data")
 args=parser.parse_args()
 
 model=load_model(args.model)
 
 print("loading data...")
-((trainImages,trainLabels),(testImages,testLabels))=loadDataset(args.data_dir)
+with np.load(args.data) as data:
+	trainImages=data["trainImages"]
+	trainLabels=data["trainLabels"]
+	testImages=data["testImages"]
+	testLabels=data["testLabels"]
 print("done")
 
 for img in testImages: