Add import for TestProbe

The example for TestProbe doesn't seem to work very well without the TestProbe class imported. :-(
Merge pull request #15478 from kanerogers/patch-1

Add import for TestProbe