From 087b80cf8eb2ee21c3c398ee35900068882ba311 Mon Sep 17 00:00:00 2001
From: caneleex <git@mlnr.dev>
Date: Thu, 29 Apr 2021 12:35:01 +0200
Subject: [PATCH] fix another issue with buttons

---
 app/src/main/java/pl/jakubweg/ShieldButton.java | 2 +-
 app/src/main/java/pl/jakubweg/VotingButton.java | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/app/src/main/java/pl/jakubweg/ShieldButton.java b/app/src/main/java/pl/jakubweg/ShieldButton.java
index 34a703d2..b2774c3b 100644
--- a/app/src/main/java/pl/jakubweg/ShieldButton.java
+++ b/app/src/main/java/pl/jakubweg/ShieldButton.java
@@ -81,7 +81,7 @@ public class ShieldButton {
         if (_youtubeControlsLayout == null || iView == null) return;
 
         if (visible && shouldBeShown()) {
-            if (getLastKnownVideoTime() == getCurrentVideoLength()) {
+            if (getLastKnownVideoTime() >= getCurrentVideoLength()) {
                 return;
             }
             if (debug) {
diff --git a/app/src/main/java/pl/jakubweg/VotingButton.java b/app/src/main/java/pl/jakubweg/VotingButton.java
index 9795d158..215b49c7 100644
--- a/app/src/main/java/pl/jakubweg/VotingButton.java
+++ b/app/src/main/java/pl/jakubweg/VotingButton.java
@@ -81,7 +81,7 @@ public class VotingButton {
         if (_youtubeControlsLayout == null || iView == null) return;
 
         if (visible && shouldBeShown()) {
-            if (getLastKnownVideoTime() == getCurrentVideoLength()) {
+            if (getLastKnownVideoTime() >= getCurrentVideoLength()) {
                 return;
             }
             if (debug) {