Add forgotten imports for exception handling