Tidy up the error handling