modify, fix bug