diff --git a/docs/examples/ex02.py b/docs/examples/ex02.py
index 288d0d59d..0eacbc6a7 100644
--- a/docs/examples/ex02.py
+++ b/docs/examples/ex02.py
@@ -101,11 +101,8 @@ def C(T):
 K = asm(bilinf, ib)
 f = 1e6 * asm(unit_load, ib)
 
-D = np.concatenate(
-    [
-        ib.get_dofs("left").facet["u_n"],
-        *[ib.get_dofs(s).nodal["u"] for s in ["left", "right", "top"]],
-    ]
+D = np.hstack(
+    [ib.get_dofs("left").all("u_n"), ib.get_dofs({"left", "right", "top"}).all("u")]
 )
 
 x = solve(*condense(K, f, D=D))