Minor change

- Minor optimization to `data.select()` such that it matches what a JIT should produce