Fix a stupid mistake in the code preventing it from working...

The previous errors were do to a stupid mistake on my part caused during
merging.