diff --git a/inject.js b/inject.js
index 5ca7084..b15455e 100644
--- a/inject.js
+++ b/inject.js
@@ -184,7 +184,7 @@
             if(!controller){
               return;
             }
-            if (!mutation.target.currentSrc) {
+            if (!mutation.target.src && !mutation.target.currentSrc) {
               controller.classList.add('vsc-nosource');
             } else {
               controller.classList.remove('vsc-nosource');