diff --git a/tests/LoadTest.php b/tests/LoadTest.php
index 34f3097..a70e8cb 100644
--- a/tests/LoadTest.php
+++ b/tests/LoadTest.php
@@ -150,12 +150,11 @@ class LoadTest extends PHPUnit_Framework_TestCase
         $this->setConstants();
         Load::setAutoloadFrom(self::$file);
         $autoload = require(self::$file);
-        $this->setExpectedException('PHPUnit_Framework_Error');
-        $this->assertNotEmpty(Load::getFilePath('anton'));
+        $this->setExpectedException('PHPUnit_Framework_Error', 'Undefined index');
+        $this->assertNotEmpty(Load::getFilePath('ClassDontExist'));
     }
 
     /**
-     * @TODO: Load::autoload() needs self::$autoload = require(self::$file); after self::buildAutoload();
      * @runInSeparateProcess
      */
     public function testDebugAutoload()