Fixed a small bug in PSTH