Fix bug with code sample