Fix unit testing.