Add back removed import.