The plastic water pump and thermostat are the headline B8 failure items (an early water-pump recall replaced plastic housings with metal), along with the PCV/oil separator and, on DSG cars, the mechatronics. On earlier cars these should be resolved by now — confirm they've been addressed.
Water pump and thermostat are the known weak points
Audi S4 (B8) · mechanical