diff --git a/include/sparseir/sampling.hpp b/include/sparseir/sampling.hpp
index a493c52..53c73d4 100644
--- a/include/sparseir/sampling.hpp
+++ b/include/sparseir/sampling.hpp
@@ -85,6 +85,10 @@ class TauSampling : public AbstractSampling<S> {
         return sampling_points_;
     }
 
+    const Eigen::VectorXd& tau() const {
+        return sampling_points_;
+    }
+
     const Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic>& matrix() const override {
         return matrix_;
     }
diff --git a/test/augment.cxx b/test/augment.cxx
index ce2b6a1..9309178 100644
--- a/test/augment.cxx
+++ b/test/augment.cxx
@@ -141,8 +141,8 @@ TEST_CASE("Augmented bosonic basis") {
 
     // Create tau sampling points
     auto tau_sampling = TauSampling(basis_aug);
-    /*
     auto tau = tau_sampling.tau;
+    REQUIRE().size() == basis_aug.size());
     Eigen::VectorX<T> gtau(tau.size());
     for (size_t i = 0; i < tau.size(); ++i) {
         gtau(i) = c - exp(-tau(i) * pole) / (1 - exp(-beta * pole));
@@ -165,7 +165,6 @@ TEST_CASE("Augmented bosonic basis") {
     Eigen::VectorX<T> gtau_reconst = tau_matrix * gl_fit;
 
     REQUIRE(gtau_reconst.isApprox(gtau, 1e-14 * magn));
-    */
 }