fix small bug i introduced