Small exception fix