diff --git a/UI/src/views/prediction/GlobalPredictionView.vue b/UI/src/views/prediction/GlobalPredictionView.vue
index e033fcd..abd4fcf 100644
--- a/UI/src/views/prediction/GlobalPredictionView.vue
+++ b/UI/src/views/prediction/GlobalPredictionView.vue
@@ -121,7 +121,7 @@ const pagination = reactive({
 
 const filteredModelList = computed(() => {
   return modelList.value.filter(model => {
-    const modelTypeMatch = !filters.model_type || model.model_type === filters.model_type
+    const modelTypeMatch = !filters.model_type || model.model_type === filters.model_type.id
     return modelTypeMatch
   })
 })
diff --git a/UI/src/views/prediction/ProductPredictionView.vue b/UI/src/views/prediction/ProductPredictionView.vue
index c3e5cda..47a76a0 100644
--- a/UI/src/views/prediction/ProductPredictionView.vue
+++ b/UI/src/views/prediction/ProductPredictionView.vue
@@ -131,7 +131,7 @@ const pagination = reactive({
 const filteredModelList = computed(() => {
   return modelList.value.filter(model => {
     const productMatch = !filters.product_id || model.product_id === filters.product_id
-    const modelTypeMatch = !filters.model_type || model.model_type === filters.model_type
+    const modelTypeMatch = !filters.model_type || model.model_type === filters.model_type.id
     return productMatch && modelTypeMatch
   })
 })
diff --git a/UI/src/views/prediction/StorePredictionView.vue b/UI/src/views/prediction/StorePredictionView.vue
index 4e43719..25aad44 100644
--- a/UI/src/views/prediction/StorePredictionView.vue
+++ b/UI/src/views/prediction/StorePredictionView.vue
@@ -146,7 +146,7 @@ const modelsWithNames = computed(() => {
 const filteredModelList = computed(() => {
   return modelsWithNames.value.filter(model => {
     const storeMatch = !filters.store_id || model.store_id === filters.store_id
-    const modelTypeMatch = !filters.model_type || model.model_type === filters.model_type
+    const modelTypeMatch = !filters.model_type || model.model_type === filters.model_type.id
     return storeMatch && modelTypeMatch
   })
 })
diff --git a/prediction_history.db b/prediction_history.db
index f21bdc0..64b4d81 100644
Binary files a/prediction_history.db and b/prediction_history.db differ