Added missing type in mapping