Fix: make tests pass
v0.3.1