bug: fix bug