Removed auto import of Exception while testing and forgot to put it back