diff --git a/src/compiler/code.jl b/src/compiler/code.jl
index df24ee7a..aa5e847c 100644
--- a/src/compiler/code.jl
+++ b/src/compiler/code.jl
@@ -4,7 +4,7 @@ export @net, @ml
 
 function process_func(ex, params = [])
   @capture(shortdef(ex), (args__,) -> body_)
-  body = @> body MacroTools.flatten block liftloops graphm DataFlow.il
+  body = @> body MacroTools.flatten liftloops graphm DataFlow.il
   body = mapconst(x -> x in params ? :(self.$x) : x, body)
   return args, body
 end