diff --git a/src/dft/dft.cpp b/src/dft/dft.cpp
index 7af27d4..3b720ed 100644
--- a/src/dft/dft.cpp
+++ b/src/dft/dft.cpp
@@ -16,7 +16,7 @@ void define_dft(py::module& m) {
 
 
     // DFT class
-    py::class_<DFT<double>, std::shared_ptr<DFT<double>>>(m, "DFT", "DFT")
+    py::class_<DFT<double>, std::shared_ptr<DFT<double>>>(m, "DFT", "Dynamic Fault Tree")
         .def("nr_elements", &DFT<double>::nrElements, "Total number of elements")
         .def("nr_be", &DFT<double>::nrBasicElements, "Number of basic elements")
         .def("nr_dynamic", &DFT<double>::nrDynamicElements, "Number of dynamic elements")
diff --git a/tests/storage/test_state_generation.py b/tests/storage/test_state_generation.py
index f61b11c..153a4fd 100644
--- a/tests/storage/test_state_generation.py
+++ b/tests/storage/test_state_generation.py
@@ -67,7 +67,6 @@ def test_knuth_yao_die():
         number_states += 1
 
         valuation = generator.current_state_to_valuation()
-        print(valuation)
         assert valuation.integer_values[s_variable] <= 7
         assert generator.current_state_satisfies(upper_bound_invariant)