Minor code sample change