diff --git a/tests/waitPackagist.php b/tests/waitPackagist.php
index 17d729af..c3fa1c16 100755
--- a/tests/waitPackagist.php
+++ b/tests/waitPackagist.php
@@ -1,10 +1,13 @@
 #!/usr/bin/env php
 <?php
 
+$tag = \getenv('TRAVIS_TAG');
+$branch = $tag ? $tag : "dev-".\getenv("TRAVIS_BRANCH");
+
 while (true) {
     $json = \json_decode(\file_get_contents('https://repo.packagist.org/p/danog/madelineproto.json'), true);
-    if ($json["packages"]["danog/madelineproto"]["dev-".\getenv("TRAVIS_BRANCH")]["source"]["reference"] === \getenv('TRAVIS_COMMIT')) {
+    \sleep(1);
+    if ($json["packages"]["danog/madelineproto"][$branch]["source"]["reference"] === \getenv('TRAVIS_COMMIT')) {
         return;
     }
-    \sleep(1);
 }