diff --git a/src/htool/hmatrix/hmatrix_builder.hpp b/src/htool/hmatrix/hmatrix_builder.hpp
index 8ab0a2c..3f60c29 100644
--- a/src/htool/hmatrix/hmatrix_builder.hpp
+++ b/src/htool/hmatrix/hmatrix_builder.hpp
@@ -21,7 +21,6 @@ void declare_hmatrix_builder(py::module &m, const std::string &className) {
     //              py::arg("UPLO"));
 
     py_class.def(py::init([](underlying_type<CoefficientPrecision> epsilon, CoordinatePrecision eta, char symmetry, char UPLO, int reqrank, std::shared_ptr<VirtualLowRankGeneratorPython<CoefficientPrecision>> low_rank_strategy) {
-                     std::cout << epsilon << "\n";
                      return std::unique_ptr<Class>(new Class(epsilon, eta, symmetry, UPLO, reqrank, low_rank_strategy));
                  }),
                  py::arg("epsilon"),
diff --git a/src/htool/main.cpp b/src/htool/main.cpp
index 7329957..7f94e5a 100644
--- a/src/htool/main.cpp
+++ b/src/htool/main.cpp
@@ -78,6 +78,7 @@ PYBIND11_MODULE(Htool, m) {
     declare_hmatrix_builder<std::complex<double>, double>(m, "ComplexHMatrixBuilder");
     declare_HMatrix<std::complex<double>, double>(m, "ComplexHMatrix");
     declare_virtual_generator<std::complex<double>>(m, "ComplexVirtualGenerator", "IComplexGenerator");
+    declare_custom_VirtualLowRankGenerator<std::complex<double>>(m, "VirtualComplexLowRankGenerator");
 
     declare_distributed_operator<std::complex<double>>(m, "ComplexDistributedOperator");
     declare_distributed_operator_utility<std::complex<double>, double>(m, "Complex");
diff --git a/tests/test_ddm_solver.py b/tests/test_ddm_solver.py
index 42dcde1..f12f8d8 100644
--- a/tests/test_ddm_solver.py
+++ b/tests/test_ddm_solver.py
@@ -238,8 +238,8 @@ def test_ddm_solver(
             intersections,
             generator,
             geometry,
-            epsilon,
-            eta,
+            Htool.ClusterBuilder(),
+            Htool.ComplexHMatrixBuilder(epsilon, eta * 1.0, symmetry, UPLO),
         )
 
     solver = default_solver_builder.solver