diff --git a/swh/storage/tests/test_storage.py b/swh/storage/tests/test_storage.py
--- a/swh/storage/tests/test_storage.py
+++ b/swh/storage/tests/test_storage.py
@@ -1533,7 +1533,7 @@
                  'origin': data.origin2,
                  'date': data.date_visit2,
                  'visit': 123,
-                 'type': data.origin2['type'],
+                 'type': data.type_visit2,
                  'status': 'full',
                  'metadata': None,
                  'snapshot': None,
@@ -1542,7 +1542,7 @@
                  'origin': data.origin2,
                  'date': '2018-01-01 23:00:00+00',
                  'visit': 1234,
-                 'type': data.origin2['type'],
+                 'type': data.type_visit2,
                  'status': 'full',
                  'metadata': None,
                  'snapshot': None,
@@ -1557,7 +1557,7 @@
                 'origin': origin_id,
                 'date': data.date_visit2,
                 'visit': 123,
-                'type': data.origin2['type'],
+                'type': data.type_visit2,
                 'status': 'full',
                 'metadata': None,
                 'snapshot': None,
@@ -1566,7 +1566,7 @@
                 'origin': origin_id,
                 'date': data.date_visit3,
                 'visit': 1234,
-                'type': data.origin2['type'],
+                'type': data.type_visit2,
                 'status': 'full',
                 'metadata': None,
                 'snapshot': None,
@@ -1578,7 +1578,7 @@
             'origin': expected_origin,
             'date': data.date_visit2,
             'visit': 123,
-            'type': data.origin2['type'],
+            'type': data.type_visit2,
             'status': 'full',
             'metadata': None,
             'snapshot': None,
@@ -1587,7 +1587,7 @@
             'origin': expected_origin,
             'date': data.date_visit3,
             'visit': 1234,
-            'type': data.origin2['type'],
+            'type': data.type_visit2,
             'status': 'full',
             'metadata': None,
             'snapshot': None,