Add missing try-catch