diff --git a/test/Twig/Tests/Loader/ChainTest.php b/test/Twig/Tests/Loader/ChainTest.php
index 260a89d3b..927318762 100644
--- a/test/Twig/Tests/Loader/ChainTest.php
+++ b/test/Twig/Tests/Loader/ChainTest.php
@@ -73,7 +73,7 @@ class Twig_Tests_Loader_ChainTest extends PHPUnit_Framework_TestCase
 
     public function testExists()
     {
-        $loader1 = $this->getMockBuilder('Twig_Loader_Array')->setMethods(array('exists', 'getSourceContext'))->disableOriginalConstructor()->getMock();
+        $loader1 = $this->getMockBuilder('Twig_LoaderInterface')->getMock();
         $loader1->expects($this->once())->method('exists')->will($this->returnValue(false));
         $loader1->expects($this->never())->method('getSourceContext');
 
diff --git a/test/Twig/Tests/ParserTest.php b/test/Twig/Tests/ParserTest.php
index 2a4268997..05ca3e3f8 100644
--- a/test/Twig/Tests/ParserTest.php
+++ b/test/Twig/Tests/ParserTest.php
@@ -155,12 +155,9 @@ EOF
         $parser = new Twig_Parser(new Twig_Environment($this->getMockBuilder('Twig_LoaderInterface')->getMock()));
         $parser->setParent(new Twig_Node());
 
-        $stream = $this->getMockBuilder('Twig_TokenStream')->disableOriginalConstructor()->getMock();
-        $stream->expects($this->any())->method('getSourceContext')->will($this->returnValue(new Twig_Source('', '')));
-
         $p = new ReflectionProperty($parser, 'stream');
         $p->setAccessible(true);
-        $p->setValue($parser, $stream);
+        $p->setValue($parser, new Twig_TokenStream(array()));
 
         return $parser;
     }