Minor code fix.

Minor code fix in an example...