add try except on test run