diff --git a/compiler/heptagon/transformations/switch.ml b/compiler/heptagon/transformations/switch.ml
index 6d58091..8ebaabe 100644
--- a/compiler/heptagon/transformations/switch.ml
+++ b/compiler/heptagon/transformations/switch.ml
@@ -132,8 +132,8 @@ let level_up defnames constr h =
 let add_to_locals vd_env locals h =
   let add_one n nn (locals,vd_env) =
     let orig_vd = Idents.Env.find n vd_env in
-    let vd_nn = mk_var_dec nn orig_vd.v_type in
-    vd_nn::locals, Idents.Env.add vd_nn.v_ident vd_nn vd_env 
+    let vd_nn = mk_var_dec ~linearity:orig_vd.v_linearity nn orig_vd.v_type in
+    vd_nn::locals, Idents.Env.add vd_nn.v_ident vd_nn vd_env
   in
     fold add_one h (locals, vd_env)
 end