Fix a missing import; #560