another small bug I overlooked when updating this class