Fix catch block