diff --git a/src/Tests/MigrateUpgradeTestBase.php b/src/Tests/MigrateUpgradeTestBase.php
index 527b3d8..e4147b4 100644
--- a/src/Tests/MigrateUpgradeTestBase.php
+++ b/src/Tests/MigrateUpgradeTestBase.php
@@ -160,6 +160,10 @@ abstract class MigrateUpgradeTestBase extends WebTestBase {
     $this->drupalPostForm(NULL, [], t('Perform rollback'));
     $this->assertText('Rollback of the upgrade is complete - you may now start the upgrade process from scratch.');
 
+    // Have to reset all the statics after rollback to ensure entities are
+    // loadable.
+    $this->resetAll();
+
     foreach (array_keys(\Drupal::entityTypeManager()->getDefinitions()) as $entity_type) {
       $real_count = count(\Drupal::entityTypeManager()->getStorage($entity_type)->loadMultiple());
       $expected_count = isset($original_counts[$entity_type]) ? $original_counts[$entity_type] : 0;