fixed import error from refactoring