diff --git a/test/Test.hx b/test/Test.hx
index 888c9f6..d8ab884 100644
--- a/test/Test.hx
+++ b/test/Test.hx
@@ -517,10 +517,10 @@ class Test extends haxe.unit.TestCase {
 	}
 
 	function testArrowFunctions() {
-		peq("class C { var f = () -> Math.random(); }", "class C {var f = function() return Math.random();}");
-		peq("class C { var f = (i) -> i * i + 2; }", "class C {var f = function(i) return i * i + 2;}");
-		peq("class C { var f = (i:Int) -> i * i + 2; }", "class C {var f = function(i:Int) return i * i + 2;}");
-		peq("class C { var f = (i:Int, j:Float) -> i * j + 2; }", "class C {var f = function(i:Int, j:Float) return i * j + 2;}");
+		peq("class C { var f = () -> Math.random(); }", "class C {var f = () -> return Math.random();}");
+		peq("class C { var f = (i) -> i * i + 2; }", "class C {var f = i -> return i * i + 2;}");
+		peq("class C { var f = (i:Int) -> i * i + 2; }", "class C {var f = (i:Int) -> return i * i + 2;}");
+		peq("class C { var f = (i:Int, j:Float) -> i * j + 2; }", "class C {var f = (i:Int, j:Float) -> return i * j + 2;}");
 		peq("class C { var f:Int -> Int; }", "class C {var f : Int -> Int;}");
 		peq("class C { var f:Int -> Float -> Int; }", "class C {var f : (Int, Float) -> Int;}");
 		peq("class C { var f:(i:Int) -> Int; }", "class C {var f : (i:Int) -> Int;}");