add mapping for comp_stage