Fix broken exception import