Fix readme code sample