diff --git a/src/BatchProcessor.php b/src/BatchProcessor.php
index fbf12af472a0a81883970712591f0194125fc25c..418df2d480c858652f4ec126cfd99e61b3ba1459 100644
--- a/src/BatchProcessor.php
+++ b/src/BatchProcessor.php
@@ -2,8 +2,8 @@
 
 namespace Drupal\automatic_updates;
 
-use Drupal\automatic_updates\Exception\UpdateException;
 use Drupal\Core\Url;
+use Drupal\package_manager\Exception\StageValidationException;
 use Symfony\Component\HttpFoundation\RedirectResponse;
 
 /**
@@ -38,7 +38,7 @@ class BatchProcessor {
       $error->getMessage(),
     ];
 
-    if ($error instanceof UpdateException) {
+    if ($error instanceof StageValidationException) {
       foreach ($error->getResults() as $result) {
         $messages = $result->getMessages();
         if (count($messages) > 1) {