Add missing import.
Add test_probe_for_id.