1 files changed, 2 insertions, 2 deletions
diff --git a/src/solvers.rs b/src/solvers.rs
index 0c01042..9bf0ad6 100644
--- a/src/solvers.rs
+++ b/src/solvers.rs
@@ -18,9 +18,9 @@ impl Wall {
     }
 
     pub fn calculate_row(&self, r: u32, stones: &mut [u32]) {
-        let mut len = 0;
+        let mut len = 1;
         let mut i = 0;
-        for &height in self.heights.iter() {
+        for &height in self.heights.iter().chain([r].iter()) {
             if height == r {
                 stones[i] = len;
                 i += 1;