Fixing a small mistake in example code