Can you change this to just `Exception`?