fix: correct `NoneType` object error

[Sanmilee]

Description

Thank you for opening a Pull Request! Before submitting your PR, there are a few things you can do to make sure it goes smoothly:

• [x] Follow the CONTRIBUTING Guide.
• [x] You are listed as the author in your notebook or README file.
  • [ ] Your account is listed in CODEOWNERS for the file(s).
• [x] Make your Pull Request title in the https://www.conventionalcommits.org/ specification.
• [x] Ensure the tests and linter pass (Run nox -s format from the repository root to format).
• [x] Appropriate docs were updated (if necessary)

Fixes #<issue_number_goes_here> 🦕

[Sanmilee]

@holtskinner, thanks for your prompt review. The PR has been revised!

[holtskinner]

@koverholt Can you please review this?

[koverholt]

Great stuff, thanks for your contribution! At first glance this all looks great, I'll give this a full review and test later today.