Whoops missed stub fix