Got rid of try except block