chore: update return type