fix reference in data categories upgrade migration (#1223)

ethyca · Aug 30, 2022 · be3b6b4 · be3b6b4
1 parent 0c55f6b
commit be3b6b4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/fidesops/ops/migrations/versions/7abe778b7082_update_fideslang_data_categories.py b/src/fidesops/ops/migrations/versions/7abe778b7082_update_fideslang_data_categories.py
@@ -178,7 +178,8 @@ def run_migration(migration_direction: str) -> None:
                 if counter % 10 == 0:
                     logger.info("Upgrading fideslang categories")
 
-                collections = dataset.get("collections")
+                dataset_json = dataset.dataset
+                collections = dataset_json.get("collections")
 
                 if collections:
                     for collection in collections: