fix small bug in last patch