a small modify