Add missing error return