diff --git a/src/extra.lua b/src/extra.lua
index 2214fd4..33a197c 100644
--- a/src/extra.lua
+++ b/src/extra.lua
@@ -1,9 +1,7 @@
 
--- @TODO make it work with functions that return multiple values
--- right now it discards returned values beyond the first
 function fprofile(f, ...)
     local t1 = am.current_time()
-    local result = f(...)
+    local result = { f(...) }
     log("%f", am.current_time() - t1)
     return result
 end