Correct slight mistake in output