diff --git a/Cargo.toml b/Cargo.toml
index 4c2f4a2..257578f 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "boolean-network-sketches"
-version = "0.1.0"
+version = "0.2.0"
 authors = ["Ondřej Huvar <xhuvar@fi.muni.cz>", "Samuel Pastva <sam.pastva@gmail.com>"]
 edition = "2021"
 description = "Framework for Boolean network inference using BN sketches."
diff --git a/src/main.rs b/src/main.rs
index 4f469f8..6d5d803 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -149,12 +149,12 @@ pub fn run_inference(
         println!("SUMMARIZING ALL CONSISTENT CANDIDATES\n");
         println!("There are following variants of update functions for each variable:");
         summarize_candidates_naively(&graph, valid_colors.clone(), true);
-    }
 
-    if n_witnesses > 0 {
-        println!("\n---------------------------------\nGENERATING WITNESSES\n");
-    } else {
-        println!("\n---------------------------------\n");
+        if n_witnesses > 0 {
+            println!("\n---------------------------------\nGENERATING WITNESSES\n");
+        } else {
+            println!("\n---------------------------------\n");
+        }
     }
 
     // generate random witnesses if required