Fix in test mapping