Attempt to fix InvocationError on test.