I took a brief look on the current code-base, and noticed that there're no unit tests!
Adding ones would help us better maintain this project in longer term.
Also, I'm happy to contribute here.
Please let me know if there's specific type/func to which you'd like to add tests.