diff --git a/tests/RepositoryTest.php b/tests/RepositoryTest.php
index 65cf9f5..b59be0c 100644
--- a/tests/RepositoryTest.php
+++ b/tests/RepositoryTest.php
@@ -9,8 +9,8 @@ class RepositoryTest extends TestCase
 {
     public function testIsSanitizingSearchWithPager()
     {
-        $client = $this->prophesize(Client::class);
-        $client->run(Argument::type(Repository::class), "grep -i --line-number -- '=sleep 5;' master")->shouldBeCalled();
+        $client = $this->prophesize('GitList\Git\Client');
+        $client->run(Argument::type('GitList\Git\Repository'), "grep -i --line-number -- '=sleep 5;' master")->shouldBeCalled();
 
         $repository = new Repository('/tmp', $client->reveal());
         $repository->searchTree('--open-files-in-pager=sleep 5;', 'master');
@@ -19,8 +19,8 @@ class RepositoryTest extends TestCase
 
     public function testIsSanitizingSearchWithAnyOption()
     {
-        $client = $this->prophesize(Client::class);
-        $client->run(Argument::type(Repository::class), "grep -i --line-number -- 'foobar  =bar;' foo")->shouldBeCalled();
+        $client = $this->prophesize('GitList\Git\Client');
+        $client->run(Argument::type('GitList\Git\Repository'), "grep -i --line-number -- 'foobar  =bar;' foo")->shouldBeCalled();
 
         $repository = new Repository('/tmp', $client->reveal());
         $repository->searchTree('foobar --bar --foo=bar;', 'foo');