diff --git a/tetri5/entity.py b/tetri5/entity.py index 0a2a5e8..069f190 100644 --- a/tetri5/entity.py +++ b/tetri5/entity.py @@ -354,6 +354,10 @@ class Stack(Entity): self.total_lines += self.lines_completed_last def draw(self, surface: pygame.Surface) -> None: + # only draw if squares and colors are aligned + if len(self._square_colors) != len(self._points): + return + for i in range(len(self._points)): square = self._points[i] square_design = self._square_colors[i]