Fixes on the code sample