diff --git a/app/src/main/java/pl/jakubweg/ShieldButton.java b/app/src/main/java/pl/jakubweg/ShieldButton.java
index 34a703d..b2774c3 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 9795d15..215b49c 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) {