Add the probe test back