Add a more proper callback for diagnostic responses.