Fixed a mistake in example code