Fix tests for stub.