BUG: avoid same indices in test and train