Update import after package refactoring.