fix mistake in plotting example