Fix spiketrain issue