Fix: a :bug: in test/train split slicing