Fixed code snippet

To make it look close to Lua code :)