diff --git a/src/Plugin/Purge/DiagnosticCheck/AcquiaCloudCheck.php b/src/Plugin/Purge/DiagnosticCheck/AcquiaCloudCheck.php
index 552dfdd..1a394c6 100644
--- a/src/Plugin/Purge/DiagnosticCheck/AcquiaCloudCheck.php
+++ b/src/Plugin/Purge/DiagnosticCheck/AcquiaCloudCheck.php
@@ -60,7 +60,7 @@ class AcquiaCloudCheck extends DiagnosticCheckBase implements DiagnosticCheckInt
    * {@inheritdoc}
    */
   public function run() {
-    $version = system_get_info('module', 'acquia_purge')['version'];
+    $version = \Drupal::service('extension.list.$type')->getExtensionInfo;
     $version = is_null($version) ? '8.x-1.x-dev' : $version;
     $this->value = $version;