Perhaps you’re expecting a block matrix? If you assemble the system as a block matrix system (e.g., injecting your code at roughly this point in the demo), you should see something like:
Bear in mind that DOLFINx reorders DoFs to minimise matrix bandwidth which is important for computational efficiency.