Fixed another small oversight