Add missing exception import.