fixed minor bug from refactoring"