docs: correct code demo

There were some errors when
trying to step through the
code demo.