diff --git a/tests/Keboola/StorageApi/MaintenanceTest.php b/tests/Keboola/StorageApi/MaintenanceTest.php
index 224442fd4..fdaa5351b 100644
--- a/tests/Keboola/StorageApi/MaintenanceTest.php
+++ b/tests/Keboola/StorageApi/MaintenanceTest.php
@@ -21,9 +21,8 @@ public function testMaintenance()
 			$this->assertEquals(503, $e->getCode());
 			$params = $e->getContextParams();
 			$this->assertEquals('maintenance', $params["status"]);
-			$paramsKeys = array_keys($params);
-			$this->assertTrue(in_array("estimatedEndTime", $paramsKeys));
-			$this->assertTrue(in_array("reason", $paramsKeys));
+			$this->assertArrayHasKey('reason', $params);
+			$this->assertArrayHasKey('estimatedEndTime', $params);
 		}
 	}
 }
\ No newline at end of file