add missing trace decorator