Fix tests_stubs tests
Fix tests_stubs tests

Minor fixes for api changes from a while ago.
Since the "tests_stubs" directory starts with "tests_", it isn't/wasn't
run automatically by nosetests when the code was changed.

beaulyddon-wf/furious,beaulyddon-wf/furious,Workiva/furious,Workiva/furious,mattsanders-wf/furious,mattsanders-wf/furious,andreleblanc-wf/furious,rosshendrickson-wf/furious,andreleblanc-wf/furious,rosshendrickson-wf/furious