a small modify.