diff --git a/augment/main.lua b/augment/main.lua
index aa25bfd..eec6fe3 100644
--- a/augment/main.lua
+++ b/augment/main.lua
@@ -38,7 +38,7 @@ for i, name in ipairs(fs.list("modules")) do
     io.write("\n")
 
     if success then
-        table.insert(modules, callback)
+        table.insert(modules, run_wrapped(callback, name))
     end
     term.setTextColor(colors.white)
 end