diff --git a/src/utils.jl b/src/utils.jl
index 7ab50f82..37a153d2 100644
--- a/src/utils.jl
+++ b/src/utils.jl
@@ -248,14 +248,14 @@ function with(f::Function, args...; kw...)
     end
   end
 
-  display(olddefs)
-  display(newdefs)
+  # display(olddefs)
+  # display(newdefs)
 
   # now set all those defaults
   default(; newdefs...)
 
   # call the function
-  f()
+  ret = f()
 
   # put the defaults back
   default(; olddefs...)
@@ -265,7 +265,8 @@ function with(f::Function, args...; kw...)
     backend(oldbackend)
   end
 
-  return
+  # return the result of the function
+  ret
 end
 
 # ---------------------------------------------------------------