Set index to 0 before saving if at end of set

diff --git a/lightcards/display.py b/lightcards/display.py
index 2fc7058f997f0dedab6af4886728d578658e449d..b4076e57a5e8ac20a6d262c43c2d467d93f75421 100644 (file)
--- a/lightcards/display.py
+++ b/lightcards/display.py
@@ -24,6 +24,9 @@ class Display():
         self.get_key()
 
     def leave(self):
         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)
 
         progress.dump((self.obj, self.stack, self.headers), self.stack)
         sys.exit(0)