[Refactor] Follow-up for PR 1930

[MortalHappiness]

Why are these changes needed?

Follow-up for https://github.com/ray-project/kuberay/pull/1930#pullrequestreview-2044141001

Related issue number

N/A

Checks

• [x] I've made sure the tests are passing.
• Testing Strategy
  • [x] Unit tests
  • [ ] Manual tests
  • [ ] This PR is not tested :(

[kevin85421]

Related to #1930