Add missing error handling.