1 files changed, 11 insertions, 2 deletions
diff --git a/crates/rocie-server/src/api/set/unit.rs b/crates/rocie-server/src/api/set/unit.rs
index 4281e05..5f39c8f 100644
--- a/crates/rocie-server/src/api/set/unit.rs
+++ b/crates/rocie-server/src/api/set/unit.rs
@@ -2,13 +2,21 @@ use actix_web::{HttpResponse, Responder, Result, post, web};
 use serde::Deserialize;
 use utoipa::ToSchema;
 
-use crate::{app::App, storage::sql::{insert::Operations, unit::{Unit, UnitId}}};
+use crate::{
+    app::App,
+    storage::sql::{
+        insert::Operations,
+        unit::{Unit, UnitId},
+        unit_property::UnitPropertyId,
+    },
+};
 
 #[derive(Deserialize, ToSchema)]
 struct UnitStub {
     full_name_plural: String,
     full_name_singular: String,
     short_name: String,
+    unit_property: UnitPropertyId,
     description: Option<String>,
 }
 
@@ -17,7 +25,7 @@ struct UnitStub {
     responses(
         (
             status = 200,
-            description = "Product successfully registered in database",
+            description = "Unit successfully registered in database",
             body = UnitId,
         ),
         (
@@ -40,6 +48,7 @@ pub(crate) async fn register_unit(
         unit.full_name_plural.clone(),
         unit.short_name.clone(),
         unit.description.clone(),
+        unit.unit_property,
         &mut ops,
     );