Fix a missing import.