Notebook corrections

[calgray]

• Closes #xxxx
• Tests added
• Passes pre-commit run --all-files
• User visible changes (including notable bug fixes) are documented in whats-new.rst
• New functions/methods are listed in api.rst

[codecov-commenter]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 93.16%. Comparing base (ec7aacd) to head (5d6743a).

Additional details and impacted files

@@           Coverage Diff           @@
##             main      #12   +/-   ##
=======================================
  Coverage   93.16%   93.16%           
=======================================
  Files          14       14           
  Lines        1653     1653           
  Branches      158      158           
=======================================
  Hits         1540     1540           
  Misses         74       74           
  Partials       39       39           

Flag	Coverage Δ
unittests	93.16% <ø> (ø)

Flags with carried forward coverage won't be shown. Click here to find out more.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.