I have the m104 engine and it has an auxilary water pump that keeps the water circulating for up to 10 mins after the car is swithched off.
It is located just below the windscreen washer container that holds the water.
Check if you have it,you should.