Fix mistake in sample code