refactor: fix bug