Fixes to KH Probe client class