slightly modify

slightly modify