Fix train samples