From 6489861dc7bd0f5ee4b7e895a8d65afcb3df8c82 Mon Sep 17 00:00:00 2001
From: Yuriy Liskov <firsthash@gmail.com>
Date: Wed, 14 Oct 2020 05:22:52 +0300
Subject: [PATCH] state on play end fix

---
 .../app/models/playback/managers/StateUpdater.java       | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/common/src/main/java/com/liskovsoft/smartyoutubetv2/common/app/models/playback/managers/StateUpdater.java b/common/src/main/java/com/liskovsoft/smartyoutubetv2/common/app/models/playback/managers/StateUpdater.java
index ebb678ed3..d3ac22089 100644
--- a/common/src/main/java/com/liskovsoft/smartyoutubetv2/common/app/models/playback/managers/StateUpdater.java
+++ b/common/src/main/java/com/liskovsoft/smartyoutubetv2/common/app/models/playback/managers/StateUpdater.java
@@ -148,6 +148,15 @@ public class StateUpdater extends PlayerEventListenerHelper {
         }
     }
 
+    @Override
+    public void onPlayEnd() {
+        Video video = mController.getVideo();
+
+        if (video != null) {
+            mStates.remove(video.videoId);
+        }
+    }
+
     private void ensureVideoSize(Video item) {
         State state = mStates.get(item.videoId);