Fix broken subsample doctest