Fix a mistake in that function I just added.