Fix failing tests

[mikebronner]

• There are a few tests that recently started failing. These need to be investigated and fixed.
• Also, the test workflow needs to be updated.

I don't believe that the failing tests are representative of logic failure in the package, rather it is probably due to intentional changes made to the logic, to which the tests no longer comply.

Happy to accept any PRs that address the failing tests, otherwise will look into it over the next few days.