Fixed mistake in last commit