Fixing small mistake in last cleanup