Typo

Statement was supposed to be VLoop

tensor_ml/loop_nest.ml

@@ -31,7 +31,7 @@ let rec map_dim f dim loop = match loop with
 
 let rec propagate_expr old_id expr = function
   | Statement stmt -> Statement (Instruction.map_expr (VarExpr.alpha_replace old_id expr) stmt)
-  | VLoop vec -> Statement (Instruction.map_expr (VarExpr.alpha_replace old_id expr) vec)
+  | VLoop vec -> VLoop (Instruction.map_expr (VarExpr.alpha_replace old_id expr) vec)
   | Unroll ({ start; body; _} as l_args) ->
     let start = VarExpr.alpha_replace old_id expr start in
     let body = List.map (propagate_expr old_id expr) body in