Fix crash from mistake in last commit