Fix tests
Version bump 0.1.1