diff --git a/src/client.rs b/src/client.rs
index 7ec7080..2cc47c2 100644
--- a/src/client.rs
+++ b/src/client.rs
@@ -26,7 +26,7 @@ impl Client {
 
         let result = (self.i, observation);
 
-        if self.i < NUM_OF_PROJECTIONS {
+        if self.i < NUM_OF_PROJECTIONS - 1 {
             self.i += 1;
         } else {
             self.i = 0;
diff --git a/src/main.rs b/src/main.rs
index 04e5053..b951eba 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -23,7 +23,7 @@ fn main() {
     let mut rng = rand::rng();
 
     let mut count: HashMap<usize, usize> = HashMap::new();
-    for i in 0..NUM_OF_PROJECTIONS {
+    for i in 0..NUM_OF_PROJECTIONS * 3 {
         let current_grid = *grid_candidates.choose(&mut rng).unwrap();
 
         let (index, observation) = client.observe_quantized(current_grid);