Fix implementation for manual indices in batch kibble