Fixed broken tests.

Closes #3.