diff --git a/lib/livebook/runtime/evaluator.ex b/lib/livebook/runtime/evaluator.ex
index ae506af53..51aa2477e 100644
--- a/lib/livebook/runtime/evaluator.ex
+++ b/lib/livebook/runtime/evaluator.ex
@@ -752,8 +752,7 @@ defmodule Livebook.Runtime.Evaluator do
             :code.load_binary(module, String.to_charlist(file), binary)
 
           # Registration of module
-          env = %{env | module: module, versioned_vars: %{}}
-          Evaluator.Tracer.trace({:on_module, binary, %{}}, env)
+          Evaluator.Tracer.trace({:on_module, binary, %{}}, %{env | module: module, versioned_vars: %{}})
 
           {{:ok, {:ok, module}, binding, env}, []}