Fix problem with new distribution diagnostics prefac.