Fix try block