Tests: fixed the stub