Description:

Please include a summary of the change and which issue is fixed. Please also include relevant motivation and context. List any dependencies that are required for this change.

Fixes #(issue)

Type of change

Please tick the appropriate option using [x]

Bug fix
New feature

Checklist:

Please tick the appropriate options using [x]

My PR includes tests that cover my code changes.
Lint and formatter run with no errors
All new and old tests are passing