Added missing exception import