Restored test code related to probe