Fixed a mistake in the last commit