perhaps use a separate TestProbe instead?