diff --git a/ui/src/features/calculation_instance/model/calculation_model.ts b/ui/src/features/calculation_instance/model/calculation_model.ts
index 18fda2e..9fd6c2d 100644
--- a/ui/src/features/calculation_instance/model/calculation_model.ts
+++ b/ui/src/features/calculation_instance/model/calculation_model.ts
@@ -6,6 +6,7 @@ export enum ModelMachineLearningTypes {
   OBJECT_DETECTION = "OBJECT_DETECTION",
   POSE_ESTIMATION = "POSE_ESTIMATION",
   BOP_DATASET = "BOP_DATASET",
+  WEIGHTS = "WEIGHTS",
 }
 
 export class CalculationModel extends ValidationModel {