Fixed a small mistake in previous commit