Test: use stub method instead of mock