diff --git a/util/test/demos/d3d11/d3d11_test.cpp b/util/test/demos/d3d11/d3d11_test.cpp
index 63eb244fa..933972ce3 100644
--- a/util/test/demos/d3d11/d3d11_test.cpp
+++ b/util/test/demos/d3d11/d3d11_test.cpp
@@ -613,14 +613,14 @@ ID3D11ComputeShaderPtr D3D11GraphicsTest::CreateCS(ID3DBlobPtr blob)
   return ret;
 }
 
-ID3D11GeometryShaderPtr D3D11GraphicsTest::CreateGS(ID3DBlobPtr blob,
-                                                    const std::vector<D3D11_SO_DECLARATION_ENTRY> &sodecl,
-                                                    const std::vector<UINT> &strides)
+ID3D11GeometryShaderPtr D3D11GraphicsTest::CreateGS(
+    ID3DBlobPtr blob, const std::vector<D3D11_SO_DECLARATION_ENTRY> &sodecl,
+    const std::vector<UINT> &strides, UINT rastStream)
 {
   ID3D11GeometryShaderPtr ret;
   CHECK_HR(dev->CreateGeometryShaderWithStreamOutput(blob->GetBufferPointer(), blob->GetBufferSize(),
                                                      &sodecl[0], (UINT)sodecl.size(), &strides[0],
-                                                     (UINT)strides.size(), 0, NULL, &ret));
+                                                     (UINT)strides.size(), rastStream, NULL, &ret));
   return ret;
 }
 
diff --git a/util/test/demos/d3d11/d3d11_test.h b/util/test/demos/d3d11/d3d11_test.h
index 10b522507..2e9fa28d3 100644
--- a/util/test/demos/d3d11/d3d11_test.h
+++ b/util/test/demos/d3d11/d3d11_test.h
@@ -78,7 +78,7 @@ struct D3D11GraphicsTest : public GraphicsTest
   ID3D11GeometryShaderPtr CreateGS(ID3DBlobPtr blob);
   ID3D11GeometryShaderPtr CreateGS(ID3DBlobPtr blob,
                                    const std::vector<D3D11_SO_DECLARATION_ENTRY> &sodecl,
-                                   const std::vector<UINT> &strides);
+                                   const std::vector<UINT> &strides, UINT rastStream = 0);
 
   ID3DBlobPtr SetBlobPath(std::string name, ID3DBlob *blob);
   void SetBlobPath(std::string name, ID3D11DeviceChild *shader);