Fix test.

Fixes #7.