Bug fix for single trial data