Error handling here too. Currently will catch the AssertionError, which is not what we would want.