diff --git a/docs/diagrams/container.puml b/docs/diagrams/container.puml
index 167cc3eaf..26ca790ec 100644
--- a/docs/diagrams/container.puml
+++ b/docs/diagrams/container.puml
@@ -34,8 +34,8 @@ Boundary(aws_cloudplatform, "Cloud Platform", "A Kubernetes cluster using Amazon
}
Rel(consumer, spring_boot_api, "Makes API calls to", "JSON/HTTPS")
-Rel(consumer, messaging, "Retrieves updates")
-Rel_Right(spring_boot_api, messaging, "Posts updates")
+Rel(messaging, consumer, "Retrieves updates")
+Rel_Up(upstream, messaging, "Posts updates")
Rel_Right(spring_boot_api, hmpps_auth, "Authenticates", "Bearer Token")
Rel_Down(spring_boot_api, upstream, "Makes API calls to", "JSON/HTTPS")
diff --git a/docs/diagrams/container.svg b/docs/diagrams/container.svg
index be008d3c8..7f04df85f 100644
--- a/docs/diagrams/container.svg
+++ b/docs/diagrams/container.svg
@@ -1 +1 @@
-
\ No newline at end of file
+
\ No newline at end of file
diff --git a/docs/diagrams/context.puml b/docs/diagrams/context.puml
index a5bbe6a3c..999c93549 100644
--- a/docs/diagrams/context.puml
+++ b/docs/diagrams/context.puml
@@ -21,7 +21,7 @@ Boundary(moj, "Ministry of Justice (MOJ)", "Ministerial Department") {
}
Rel_Down(consumer, hmpps_integration_api, "Uses")
-Rel_Right(hmpps_integration_api, messaging, "Posts updates")
+Rel_Up(upstream, messaging, "Posts updates")
Rel_Up(messaging, consumer, "Retrieves updates")
Rel(hmpps_integration_api, upstream, "Queries for data")
diff --git a/docs/diagrams/context.svg b/docs/diagrams/context.svg
index 0c756d007..8a2b3bf76 100644
--- a/docs/diagrams/context.svg
+++ b/docs/diagrams/context.svg
@@ -1 +1 @@
-
\ No newline at end of file
+
\ No newline at end of file