diff --git a/src/gallium/auxiliary/util/u_threaded_context.c b/src/gallium/auxiliary/util/u_threaded_context.c
index 1b4aaf28b48..8eb7f86f161 100644
--- a/src/gallium/auxiliary/util/u_threaded_context.c
+++ b/src/gallium/auxiliary/util/u_threaded_context.c
@@ -3490,6 +3490,7 @@ tc_draw_vertex_state(struct pipe_context *_pipe,
          p->state = state;
 
       take_vertex_state_ownership = false;
+      p->partial_velem_mask = partial_velem_mask;
       p->info.mode = info.mode;
       p->info.take_vertex_state_ownership = false;
       p->num_draws = dr;