small modify

small modify