fix bug from refactoring