Improve test coverage

[hypergonial]

Summary

Improve test coverage by adding more unit tests.

Why is this needed?

Decrease the likelyhood of @hypergonial messing up another release.

Ideal implementation

Write more tests.

Checklist

• I have searched the issue tracker and have made sure it's not a duplicate. If it is a follow up of another issue, I have specified it.