Add try catch to test