An assembly includes a cold plate enclosure and a flow directing baffle separable from the cold plate enclosure. The cold plate enclosure includes a cavity therein, an inlet for coolant ingress into the cavity, and an outlet for coolant egress out of the cavity. The baffle is attached to the enclosure within the cavity. The baffle includes flow directing features such that the baffle directs coolant flow through the cavity from the inlet to the outlet in a prescribed manner.