Fix failing test due to incorrect stub