Fix unit test for #2

Closes #2
Release v1.2.0