Clean up trace-replay mode switching in pspgen.