diff --git a/main.lua b/main.lua
index 5eb3d24..a7eb30f 100644
--- a/main.lua
+++ b/main.lua
@@ -154,7 +154,7 @@ local function make_network(input_size)
     nn_tz = nn_ty
     if cfg.reduce_tiles then
         nn_tz = nn_tz:feed(nn.Reshape{11, 17 * 2})
-        nn_tz = nn_tz:feed(nn.DenseBroadcast(5))
+        nn_tz = nn_tz:feed(nn.DenseBroadcast(5, true))
         nn_tz = nn_tz:feed(nn.Relu())
         -- note: due to a quirk in Merge, we don't need to flatten nn_tz.
     end