Add a wait for registration to work check to test head.