[core] Add back a missing import.