Corrected the testing mock path.

It looks like the import path changed but the test wasn't updated. This
fixes the current test failure on Travis.