2 files changed, 6 insertions, 6 deletions

diff --git a/src/luarocks/fun.lua b/src/luarocks/fun.lua
index b315711d..95f9177d 100644
--- a/src/luarocks/fun.lua
+++ b/src/luarocks/fun.lua
@@ -48,9 +48,9 @@ function fun.traverse(t, f)
    return fun.map(t, function(x)
 
       if type(x) == "table" then
-         fun.traverse(x, f)
+         return fun.traverse(x, f)
       else
-         f(x)
+         return f(x)
       end
    end)
 end
diff --git a/src/luarocks/fun.tl b/src/luarocks/fun.tl
index a63f0404..9d9551ed 100644
--- a/src/luarocks/fun.tl
+++ b/src/luarocks/fun.tl
@@ -44,13 +44,13 @@ function fun.filter<K>(xs: {K}, f: function): {K}
    return rs
 end
 
-function fun.traverse<K, V>(t: {K}, f: function(K): V): {V} --? right or {any} | any
-   return fun.map(t, function(x: K): V
+function fun.traverse<K, V>(t: {K}, f: function(K): V): {V} | V --? right or {any} | any
+   return fun.map(t, function(x: K): {V} | V
       -- return x is table and fun.traverse(x, f) or f(x) --? 
       if x is {any} then
-         fun.traverse(x, f)
+         return fun.traverse(x, f)
       else
-         f(x)
+         return f(x)
       end
    end)
 end