Fix a missing import