fix bug in tests
Merge pull request #26 from beckermr/master

fix bug in unittest script