diff --git a/data/dataset/mongo_example_test_dataset.yml b/data/dataset/mongo_example_test_dataset.yml
index ece5817b1a..587e74b317 100644
--- a/data/dataset/mongo_example_test_dataset.yml
+++ b/data/dataset/mongo_example_test_dataset.yml
@@ -7,6 +7,8 @@ dataset:
         fields:
           - name: _id
             data_categories: [system.operations]
+            fides_meta:
+              primary_key: True
           - name: customer_id
             data_categories: [user.unique_id]
             fides_meta:
@@ -203,10 +205,11 @@ dataset:
               data_type: string
           - name: id
             data_categories: [user.unique_id]
-            references:
-              - dataset: mongo_test
-                field: flights.pilots
-                direction: from
+            fides_meta:
+              references:
+                - dataset: mongo_test
+                  field: flights.pilots
+                  direction: from
           - name: name
             data_categories: [user.name]
             fides_meta:
diff --git a/tests/ops/generator/test_data_generator.py b/tests/ops/generator/test_data_generator.py
index 659185d080..d990169f28 100644
--- a/tests/ops/generator/test_data_generator.py
+++ b/tests/ops/generator/test_data_generator.py
@@ -31,6 +31,7 @@
       - name: address
         fields:
           - name: id
+            fides_meta:
               data_type: integer
           - name: user_id
           - name: street