Fix probe tests

Change-Id: I03573bf24baf031b1874c3ff2e2d89d34473c266