diff --git a/src/test/main.cpp b/src/test/main.cpp
index 4aeee185b..967435167 100644
--- a/src/test/main.cpp
+++ b/src/test/main.cpp
@@ -251,7 +251,3 @@ int main(int argc, char ** argv) {
     TST(solver_pool);
     //TST_ARGV(hs);
 }
-
-void initialize_mam() {}
-void finalize_mam() {}
-
diff --git a/src/test/prime_generator.cpp b/src/test/prime_generator.cpp
index df5fae0aa..3a63b805a 100644
--- a/src/test/prime_generator.cpp
+++ b/src/test/prime_generator.cpp
@@ -24,6 +24,7 @@ void tst_prime_generator() {
     scoped_mpz sqrt_p(m);
 
     prime_generator gen;
+    gen.initialize();
     for (unsigned i = 0; i < 10000; i++) {
         uint64_t p = gen(i);
         std::cout << p << ", ";
@@ -38,5 +39,6 @@ void tst_prime_generator() {
             ENSURE(p % i != 0);
         }
     }
-    std::cout << std::endl;    
+    gen.finalize();
+    std::cout << std::endl;
 }
diff --git a/src/util/prime_generator.cpp b/src/util/prime_generator.cpp
index 4e0afb03f..21f84e71c 100644
--- a/src/util/prime_generator.cpp
+++ b/src/util/prime_generator.cpp
@@ -129,7 +129,7 @@ uint64_t prime_iterator::next() {
 }
 
 void prime_iterator::initialize() {
-  g_prime_generator.initialize();
+    g_prime_generator.initialize();
 }
 
 void prime_iterator::finalize() {