small bug fix, customize function did not work properly