Fix tests
v0.1.2