Add back missing try catch