Correct a little mistake in last commit