Update tests.  Fixes #9.