Fix refactor_test broken import.