Minor Refactor

[UP929312]

This should clean up the logic slightly, with it being slightly more readable in my opinion, and even though it might be faster with Python's EAFP logic, I think maintainability and readability is probably more important than this micro efficiency.

Summary

Checklist

• If code changes were made then they have been tested.
  • I have updated the documentation to reflect the changes.
• If type: ignore comments were used, a comment is also left explaining why
• This PR fixes an issue.
• This PR adds something new (e.g. new method or parameters).
• This PR is a breaking change (e.g. methods or parameters removed/renamed)
• This PR is not a code change (e.g. documentation, README, typehinting, examples, ...)