diff --git a/src/entity/emr/emr_diagnosis.ts b/src/entity/emr/emr_diagnosis.ts
index 225e492..1b37ca4 100644
--- a/src/entity/emr/emr_diagnosis.ts
+++ b/src/entity/emr/emr_diagnosis.ts
@@ -13,6 +13,7 @@ import { EmrDiagnosisType } from "../../enum/emrDiagnosisType";
 import { EmrPatient } from "./emr_patient";
 import { Diagnosis } from "../diagnosis";
 import { User } from "../users";
+import { EmrRegistration } from "./emr_registration";
 
 @Entity("emr_diagnosis", { schema: "emr" })
 export class EmrDiagnosis {
@@ -23,6 +24,10 @@ export class EmrDiagnosis {
     @JoinColumn()
     patient: EmrPatient;
 
+    @ManyToOne(() => EmrRegistration, { onDelete: "CASCADE" })
+    @JoinColumn()
+    registration: EmrRegistration;
+
     @ManyToOne(() => User, { onDelete: "CASCADE" })
     @JoinColumn()
     doctor: User;
diff --git a/src/entity/emr/emr_diagnostic_procedure.ts b/src/entity/emr/emr_diagnostic_procedure.ts
index 1c28613..6fd8479 100644
--- a/src/entity/emr/emr_diagnostic_procedure.ts
+++ b/src/entity/emr/emr_diagnostic_procedure.ts
@@ -12,6 +12,7 @@ import {
 import { EmrDiagnosisType } from "../../enum/emrDiagnosisType";
 import { EmrPatient } from "./emr_patient";
 import { DiagnosticProcedure } from "../diagnostic_procedure";
+import { EmrRegistration } from "./emr_registration";
 
 @Entity("emr_diagnostic_procedure", { schema: "emr" })
 export class EmrDiagnosticProcedure {
@@ -22,14 +23,18 @@ export class EmrDiagnosticProcedure {
     @JoinColumn()
     patient: EmrPatient;
 
+    @ManyToOne(() => EmrRegistration, { onDelete: "CASCADE" })
+    @JoinColumn()
+    registration: EmrRegistration;
+
     @ManyToOne(() => DiagnosticProcedure, { onDelete: "CASCADE", nullable: true })
     @JoinColumn()
     id_diagnostic_procedure: DiagnosticProcedure;
 
-   @Column({ nullable: true, type: "text" })
+    @Column({ nullable: true, type: "text" })
     diagnostic_procedure_code: string;
 
-   @Column({ nullable: true, type: "text" })
+    @Column({ nullable: true, type: "text" })
     diagnostic_procedure: string;
 
     @Column({