Add missing mapping