Cleaned up tests.

Fixes #26