Fix mistake in exercise