diff --git a/Storage/src/StorageClient.php b/Storage/src/StorageClient.php
index d59d376b8d7..53ab0cd5524 100644
--- a/Storage/src/StorageClient.php
+++ b/Storage/src/StorageClient.php
@@ -271,7 +271,7 @@ function (array $bucket) use ($userProject) {
      * }
      * @return Bucket
      */
-    public function restore($name, $generation, array $options = [])
+    public function restore(string $name, string $generation, array $options = [])
     {
         $res = $this->connection->restoreBucket([
             'bucket' => $name,
diff --git a/Storage/tests/Unit/StorageClientTest.php b/Storage/tests/Unit/StorageClientTest.php
index a03f6e55068..531e049c019 100644
--- a/Storage/tests/Unit/StorageClientTest.php
+++ b/Storage/tests/Unit/StorageClientTest.php
@@ -92,9 +92,9 @@ public function testGetSoftDeletedBucket()
 
     public function testGetsSoftDeletedBuckets()
     {
-        $this->connection->listBuckets(Argument::allOf(
+        $this->connection->listBuckets(
             Argument::withEntry('softDeleted', true)
-        ))->willReturn([
+        )->willReturn([
             'items' => [
                 ['name' => 'bucket1']
             ]