minor fix to tests
0.1.0