fix small bug introduce with last previous commit