I'd change it to restrict the try/except code