fix

Bugfix for a stupid mistake in last commit. Definitely not TDD. Still needs test.