Fix missed import.