Added exception import that was missing