diff --git a/fftma_module/gen/lib_src/log.c b/fftma_module/gen/lib_src/log.c
index 1ef1b78..b684486 100644
--- a/fftma_module/gen/lib_src/log.c
+++ b/fftma_module/gen/lib_src/log.c
@@ -148,7 +148,7 @@ void log_log(int level, const char *file, int line, const char *fmt, ...) {
   };
 
   char* env_var = getenv("ENV");
-  if (env_var != NULL && strcmp("false", env_var) == 0) return;
+  if (env_var == NULL || strcmp("false", env_var) == 0) return;
   
   char* substr_mem = strstr(fmt, "MEM");
   char* substr_cpu = strstr(fmt, "CPU");
diff --git a/tests/performance/generation/time.py b/tests/performance/generation/time.py
index 19eb542..e7f8d1d 100644
--- a/tests/performance/generation/time.py
+++ b/tests/performance/generation/time.py
@@ -31,7 +31,7 @@ def generate(N):
     variance=3.5682389
     typ=3
 
-    k = gen(nx, ny, nz, dx, dy, dz, seed, variograms, mean, variance, typ)
+    k = gen(nx, ny, nz, dx, dy, dz, seed, variograms, mean, variance, typ, 8)
 
     duration = time.time() - start_time
     return k, duration
diff --git a/tests/stages/generation/test.py b/tests/stages/generation/test.py
index 357f542..73dbc8d 100644
--- a/tests/stages/generation/test.py
+++ b/tests/stages/generation/test.py
@@ -29,7 +29,7 @@ def generate(N):
     variance=3.5682389
     typ=3
 
-    k = gen(nx, ny, nz, dx, dy, dz, seed, variograms, mean, variance, typ)
+    k = gen(nx, ny, nz, dx, dy, dz, seed, variograms, mean, variance, typ, 8)
 
     duration = time.time() - start_time
     return k, duration