Fix in order to run test properly.