add: fix bug