Fixing the dumb diag_demo