Small bugfix from failing to test a code tweak :(