diff --git a/tests-parsimony/map.c b/tests-parsimony/map.c
index abbb090..b7cb690 100644
--- a/tests-parsimony/map.c
+++ b/tests-parsimony/map.c
@@ -1,16 +1,13 @@
-enum foo { x, y, z };
-
 @{
     toChar(foo) {
         "\""+foo+"\"";
     }
-    square(foo) {
+    p2(foo) {
         foo * foo;
     }
     c = ["earth", "sky", "water"];
 }
 
-@{print(map(square, [100, 2, 3]));}
-
-@{print(map(toChar, c));}
+@{print(map(p2, [100, 2, 3]));}
 
+@{print(map(toChar, c));}
\ No newline at end of file