Fix unitteest.