diff --git a/tests/shader_runner_d3d11.c b/tests/shader_runner_d3d11.c
index d9310497..ffc89085 100644
--- a/tests/shader_runner_d3d11.c
+++ b/tests/shader_runner_d3d11.c
@@ -620,6 +620,7 @@ static bool d3d11_runner_draw(struct shader_runner *r,
     {
         cb = create_buffer(device, D3D11_BIND_CONSTANT_BUFFER,
                 runner->r.uniform_count * sizeof(*runner->r.uniforms), 0, runner->r.uniforms);
+        ID3D11DeviceContext_VSSetConstantBuffers(context, 0, 1, &cb);
         ID3D11DeviceContext_PSSetConstantBuffers(context, 0, 1, &cb);
     }
 
diff --git a/tests/shader_runner_d3d9.c b/tests/shader_runner_d3d9.c
index 4ddb8ffe..e4e1c643 100644
--- a/tests/shader_runner_d3d9.c
+++ b/tests/shader_runner_d3d9.c
@@ -345,6 +345,10 @@ static bool d3d9_runner_draw(struct shader_runner *r,
 
     if (runner->r.uniform_count)
     {
+        hr = IDirect3DDevice9_SetVertexShaderConstantF(device, 0,
+                (const float *)runner->r.uniforms, runner->r.uniform_count / 4);
+        ok(hr == D3D_OK, "Failed to set uniforms, hr %#lx.\n", hr);
+
         hr = IDirect3DDevice9_SetPixelShaderConstantF(device, 0,
                 (const float *)runner->r.uniforms, runner->r.uniform_count / 4);
         ok(hr == D3D_OK, "Failed to set uniforms, hr %#lx.\n", hr);