Improve test coverage:

Statements: 66.67% (58 / 87)
Branches: 21.05% (4 / 19)
Functions: 76.92% (20 / 26)
Lines: 66.67% (58 / 87)