Add missing line return