diff --git a/modules/cloud_service_providers/aws_cloud/tests/src/Traits/AwsCloudTestMockTrait.php b/modules/cloud_service_providers/aws_cloud/tests/src/Traits/AwsCloudTestMockTrait.php
index 9a229be9..fe3bfc47 100644
--- a/modules/cloud_service_providers/aws_cloud/tests/src/Traits/AwsCloudTestMockTrait.php
+++ b/modules/cloud_service_providers/aws_cloud/tests/src/Traits/AwsCloudTestMockTrait.php
@@ -585,7 +585,7 @@ trait AwsCloudTestMockTrait {
   /**
    * Update the volume state and instance.
    *
-   * @param array $api
+   * @param string $api
    *   Api data.
    * @param string $device
    *   The device.
@@ -596,7 +596,7 @@ trait AwsCloudTestMockTrait {
    * @param string $state
    *   The state.
    */
-  protected function updateAttachDetachVolumeMockData(array $api, $device, $volume_id, $instance_id, $state) {
+  protected function updateAttachDetachVolumeMockData($api, $device, $volume_id, $instance_id, $state) {
     $mock_data = $this->getMockDataFromConfig();
     $mock_data[$api] = [
       'Device' => $device,