fix unit tests for #198