Why the new class? Couldn't you just query on the id in the tests like we were doing before? You're using `Y.one()` which will only pick the first one selected anyways.