Fix mistake in previous commit