Fix forgotten import.