Fix missing import in new / refactored file.