Fix: tweak code sample