From d509c1b8a05e7ac88b19558fa2a7b2a04e5d04f4 Mon Sep 17 00:00:00 2001
From: Armaan Bhojwani <me@armaanb.net>
Date: Mon, 1 Feb 2021 10:17:33 -0500
Subject: [PATCH] Set index to 0 before saving if at end of set

---
 lightcards/display.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/lightcards/display.py b/lightcards/display.py
index 2fc7058..b4076e5 100644
--- a/lightcards/display.py
+++ b/lightcards/display.py
@@ -24,6 +24,9 @@ class Display():
         self.get_key()
 
     def leave(self):
+        if self.obj.getIdx() == len(self.stack):
+            self.obj.setIdx(0)
+
         progress.dump((self.obj, self.stack, self.headers), self.stack)
         sys.exit(0)
 
-- 
2.39.2