diff --git a/tests/Functional/Net/SSH2AgentTest.php b/tests/Functional/Net/SSH2AgentTest.php
index 971eaf9f..ea10110e 100644
--- a/tests/Functional/Net/SSH2AgentTest.php
+++ b/tests/Functional/Net/SSH2AgentTest.php
@@ -7,6 +7,7 @@
  */
 
 use phpseclib\Net\SSH2;
+use phpseclib\System\SSH\Agent;
 
 class Functional_Net_SSH2AgentTest extends PhpseclibFunctionalTestCase
 {
@@ -23,7 +24,7 @@ class Functional_Net_SSH2AgentTest extends PhpseclibFunctionalTestCase
     public function testAgentLogin()
     {
         $ssh = new SSH2($this->getEnv('SSH_HOSTNAME'));
-        $agent = new System_SSH_Agent;
+        $agent = new Agent;
 
         $this->assertTrue(
             $ssh->login($this->getEnv('SSH_USERNAME'), $agent),