Fix test.

Closes #355.