Fix test.

Fixes broken tests for #188