Diego had to go to time-out