diff --git a/docs/index.md b/docs/index.md
index 892093a1b7146aa7719e0e60e7701d690a3d26cc..e5315634cfe19f81de2a3a3ad70772ee8bc2025e 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -32,10 +32,12 @@ import sympy as sp
 
 from pystencils import fields, kernel
 
-from pystencilssfg import SourceFileGenerator
+from pystencilssfg import SourceFileGenerator, SfgComposer
 from pystencilssfg.source_concepts.cpp import mdspan_ref
 
-with SourceFileGenerator() as sfg:
+with SourceFileGenerator() as ctx:
+    sfg = SfgComposer(ctx)
+
     u_src, u_dst, f = fields("u_src, u_dst, f(1) : double[2D]", layout="fzyx")
     h = sp.Symbol("h")