diff --git a/src/main/java/com/ffii/fpsms/modules/master/entity/Material.kt b/src/main/java/com/ffii/fpsms/modules/master/entity/Material.kt
index c98e8a0..5e84959 100644
--- a/src/main/java/com/ffii/fpsms/modules/master/entity/Material.kt
+++ b/src/main/java/com/ffii/fpsms/modules/master/entity/Material.kt
@@ -35,10 +35,10 @@ open class Material : BaseEntity<Long>() {
     @OneToMany(mappedBy = "material", cascade = [CascadeType.ALL], orphanRemoval = true)
     @JsonManagedReference("material-uom")
     @JsonInclude(JsonInclude.Include.NON_NULL)
-    private val uom: Set<ProductMaterialWeightUnit> = HashSet()
+    private val uom: Set<ProductMaterialUom> = HashSet()
 
     @OneToMany(mappedBy = "material", cascade = [CascadeType.ALL], orphanRemoval = true)
-    @JsonManagedReference("material-uom")
+    @JsonManagedReference("material-type")
     @JsonInclude(JsonInclude.Include.NON_NULL)
     private val type: Set<ProductMaterialType> = HashSet()