diff --git a/main.py b/main.py
index 9e9bd70..c8c7c42 100644
--- a/main.py
+++ b/main.py
@@ -289,11 +289,11 @@ def my_clip_train(in_path, out_path, model_name, source, in_base,
 	infile.close()
 
 	# evaluate with novel compositions
-	evaluations['mare_novel_comp'].append(memory_evaluation(args.in_path, 'novel_test/', memory,
+	evaluations['mare_novel_comp'].append(memory_evaluation(args.in_path, 'test/', memory,
 						bn_n_test, ['rgba'], dic_train, vocabs))
 	
 	# evaluate with variations (dic_test)
-	evaluations['mare_var'].append(memory_evaluation(args.in_path, 'novel_test/', memory,
+	evaluations['mare_var'].append(memory_evaluation(args.in_path, 'test/', memory,
 						bn_n_train, ['rgba'], dic_test, vocabs))
 	pprint(evaluations)
 	with open(os.path.join(args.out_path, evaluations), 'w') as json_file: