Fix unitests.