BUG: fix unit tests.