diff --git a/web/src/views/classification/ModelSelectionView.tsx b/web/src/views/classification/ModelSelectionView.tsx
index 2dd7e0375..b91ea5648 100644
--- a/web/src/views/classification/ModelSelectionView.tsx
+++ b/web/src/views/classification/ModelSelectionView.tsx
@@ -38,7 +38,11 @@ export default function ModelSelectionView({
return (
{classificationConfigs.map((config) => (
- onClick(config)} />
+ onClick(config)}
+ />
))}
);
diff --git a/web/src/views/classification/ModelTrainingView.tsx b/web/src/views/classification/ModelTrainingView.tsx
index 391bc90d7..d3fd5b40e 100644
--- a/web/src/views/classification/ModelTrainingView.tsx
+++ b/web/src/views/classification/ModelTrainingView.tsx
@@ -801,6 +801,7 @@ function StateTrainGrid({
>
{trainData?.map((data) => (