Consolidate driver_probe_done() loops into one place