update tests. Fixes #8.