Make tests look for classes not ids