Fixed a mistake in my last commit