small mistake

Fix small mistake in last commit