add: fix bugs