diff --git a/src/lib.rs b/src/lib.rs index a6cf45a..53b3357 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -626,8 +626,9 @@ impl Grid { } let mut residue: Vec = vec![]; for i in 0..self.rows { - self.cols -= 1; + residue.push(self.data.remove(i * self.cols + col_index - i)); } + self.cols -= 1; Some(residue) }