chalasr
diff --git a/‎src/Client.php
Copy file name to clipboardExpand all lines: src/Client.php
+4-3Lines changed: 4 additions & 3 deletions b/‎src/Client.php
Copy file name to clipboardExpand all lines: src/Client.php
+4-3Lines changed: 4 additions & 3 deletions
diff --git a/‎tests/ClientTest.php
Copy file name to clipboardExpand all lines: tests/ClientTest.php
+1-2Lines changed: 1 addition & 2 deletions b/‎tests/ClientTest.php
Copy file name to clipboardExpand all lines: tests/ClientTest.php
+1-2Lines changed: 1 addition & 2 deletions
@@ -16,7 +16,6 @@
 use Facebook\WebDriver\Exception\NoSuchElementException;
 use Facebook\WebDriver\Exception\TimeOutException;
 use Facebook\WebDriver\JavaScriptExecutor;
-use Facebook\WebDriver\Remote\RemoteWebElement;
 use Facebook\WebDriver\WebDriver;
 use Facebook\WebDriver\WebDriverBy;
 use Facebook\WebDriver\WebDriverCapabilities;
@@ -251,13 +250,15 @@ public function getCookieJar()
      * @throws NoSuchElementException
      * @throws TimeOutException
      *
-     * @return RemoteWebElement
+     * @return Crawler
      */
     public function waitFor(string $cssSelector, int $timeoutInSecond = 30, int $intervalInMillisecond = 250)
     {
-        return $this->wait($timeoutInSecond, $intervalInMillisecond)->until(
+        $this->wait($timeoutInSecond, $intervalInMillisecond)->until(
             WebDriverExpectedCondition::visibilityOfElementLocated(WebDriverBy::cssSelector($cssSelector))
         );
+
+        return $this->crawler = $this->createCrawler();
     }
 
     public function getWebDriver(): WebDriver
 
@@ -14,7 +14,6 @@
 namespace Symfony\Component\Panther\Tests;
 
 use Facebook\WebDriver\JavaScriptExecutor;
-use Facebook\WebDriver\Remote\RemoteWebElement;
 use Facebook\WebDriver\WebDriver;
 use Facebook\WebDriver\WebDriverExpectedCondition;
 use Symfony\Component\BrowserKit\Client as BrowserKitClient;
@@ -45,7 +44,7 @@ public function testWaitFor()
         $client = self::createPantherClient();
         $crawler = $client->request('GET', '/waitfor.html');
         $c = $client->waitFor('#hello');
-        $this->assertInstanceOf(RemoteWebElement::class, $c);
+        $this->assertInstanceOf(Crawler::class, $c);
         $this->assertSame('Hello', $crawler->filter('#hello')->text());
     }