minor import name bugfix due to recent refactoring.