Fix for Geweke diagnostic bug in #461