diff --git a/tests/HandlerTest.php b/tests/HandlerTest.php
index 2504d56..03de2fb 100644
--- a/tests/HandlerTest.php
+++ b/tests/HandlerTest.php
@@ -82,7 +82,7 @@
     $mock = new MockHandler();
 
     $mock->append(function (CommandInterface $cmd, RequestInterface $req) {
-        return new  ApiGatewayManagementApiException('', $cmd, ['code' => 'GoneException']);
+        return new  ApiGatewayManagementApiException('', $cmd, ['code' => 'Gone']);
     });
 
     /** @var SubscriptionRepository */