diff --git a/datahub-classify/tests/expected_output/expected_infotypes_UNIT_TESTING.json b/datahub-classify/tests/expected_output/expected_infotypes_UNIT_TESTING.json
index 65a4949..36bdc1a 100644
--- a/datahub-classify/tests/expected_output/expected_infotypes_UNIT_TESTING.json
+++ b/datahub-classify/tests/expected_output/expected_infotypes_UNIT_TESTING.json
@@ -52,7 +52,7 @@
         "General Manager": "Full_Name"
     },
     "2018-seattle-business-districts": {
-        "Address": "Street_Address",
+        "Address": "no_infotype",
         "E-mail": "Email_Address"
     },
     "Customer_Segmentation": {
@@ -129,7 +129,7 @@
         "Full_Name": "Full_Name"
     },
     "1-MB-Test": {
-        "First and Last Name": "no_infotype",
+        "First and Last Name": "Full_Name",
         "SSN" :"US_Social_Security_Number",
         "Credit Card Number" : "Credit_Debit_Card_Number"
     },
diff --git a/datahub-classify/tests/expected_output/expected_infotypes_confidence_slabs.json b/datahub-classify/tests/expected_output/expected_infotypes_confidence_slabs.json
index 4220bf3..693c5fa 100644
--- a/datahub-classify/tests/expected_output/expected_infotypes_confidence_slabs.json
+++ b/datahub-classify/tests/expected_output/expected_infotypes_confidence_slabs.json
@@ -54,7 +54,6 @@
         "General Manager": 0.6
     },
     "2018-seattle-business-districts": {
-        "Address": 0.6,
         "E-mail": 1.0
     },
     "Customer_Segmentation": {
@@ -129,6 +128,7 @@
         "Full_Name": 0.9
     },
     "1-MB-Test": {
+        "First and Last Name": 0.7,
         "SSN": 1.0,
         "Credit Card Number": 1.0
     },