diff --git a/clients/admin-ui/src/features/data-discovery-and-detection/ClassificationCategoryBadge.tsx b/clients/admin-ui/src/features/data-discovery-and-detection/ClassificationCategoryBadge.tsx
index 7d7c93b293..9cc631e799 100644
--- a/clients/admin-ui/src/features/data-discovery-and-detection/ClassificationCategoryBadge.tsx
+++ b/clients/admin-ui/src/features/data-discovery-and-detection/ClassificationCategoryBadge.tsx
@@ -15,6 +15,9 @@ const ClassificationCategoryBadge = ({
       variant="taxonomy"
       textTransform="none"
       fontWeight="normal"
+      display="flex"
+      alignItems="center"
+      gap={1.5}
       as={onClick ? "button" : undefined}
       onClick={onClick}
       _hover={onClick ? { borderColor: "primary.900" } : undefined}
diff --git a/tests/ops/integration_tests/setup_scripts/postgres_setup.py b/tests/ops/integration_tests/setup_scripts/postgres_setup.py
index e0cf34ab35..3eabf07d45 100644
--- a/tests/ops/integration_tests/setup_scripts/postgres_setup.py
+++ b/tests/ops/integration_tests/setup_scripts/postgres_setup.py
@@ -18,7 +18,7 @@
 from fides.api.models.detection_discovery import MonitorConfig
 from fides.api.models.experience_notices import ExperienceNotices
 from fides.api.models.privacy_experience import PrivacyExperienceConfig
-from fides.api.service.connectors.sql_connector import PostgreSQLConnector
+from fides.api.service.connectors.postgres_connector import PostgreSQLConnector
 from fides.config import CONFIG
 
 integration_config = load_toml("tests/ops/integration_test_config.toml")