diff --git a/test/augment.cxx b/test/augment.cxx
index 82b50cb..ce2b6a1 100644
--- a/test/augment.cxx
+++ b/test/augment.cxx
@@ -139,9 +139,9 @@ TEST_CASE("Augmented bosonic basis") {
     T pole = 1.0;
     T c = 1e-2;
 
-    /*
     // Create tau sampling points
-    auto tau_sampling = make_tau_sampling(basis_aug);
+    auto tau_sampling = TauSampling(basis_aug);
+    /*
     auto tau = tau_sampling.tau;
     Eigen::VectorX<T> gtau(tau.size());
     for (size_t i = 0; i < tau.size(); ++i) {
diff --git a/test/sampling.cxx b/test/sampling.cxx
index 5cfd5b6..39a842e 100644
--- a/test/sampling.cxx
+++ b/test/sampling.cxx
@@ -39,7 +39,7 @@ TEST_CASE("sampling", "[Sampling]") {
         Eigen::MatrixXd reconstructed_A = svd.matrixU() * svd.singularValues().asDiagonal() * svd.matrixV().transpose();
 
         // Fails
-        //REQUIRE((A - reconstructed_A).norm() <= 1e-15 * norm_A);
+        REQUIRE((A - reconstructed_A).norm() <= 1e-15 * norm_A);
 
         // Test multiplication with vector x
         Eigen::VectorXd x = Eigen::VectorXd::NullaryExpr(A.cols(), [&](auto) { return dist(rng); });