Fixed some bugs in the probe class