Fix try catch