Correct small mistake in implementation.