diff --git a/patch/ubsan/ubsan_test.patch b/patch/ubsan/ubsan_test.patch
index 53d2d1b..1c3189d 100644
--- a/patch/ubsan/ubsan_test.patch
+++ b/patch/ubsan/ubsan_test.patch
@@ -2,8 +2,8 @@
 index cae55f314..481050a43 100644
 --- a/bl31/bl31_main.c
 +++ b/bl31/bl31_main.c
-@@ -108,6 +108,19 @@ void bl31_setup(u_register_t arg0, u_register_t arg1, u_register_t arg2,
- #endif /* CTX_INCLUDE_PAUTH_REGS */
+@@ -112,6 +112,19 @@ void bl31_setup(u_register_t arg0, u_register_t arg1, u_register_t arg2,
+ #endif /* PLATFORM_REPORT_CTX_MEM_USE */
  }
  
 +/*
@@ -22,7 +22,7 @@
  /*******************************************************************************
   * BL31 is responsible for setting up the runtime services for the primary cpu
   * before passing control to the bootloader or an Operating System. This
-@@ -216,6 +229,10 @@ void bl31_main(void)
+@@ -223,6 +236,10 @@ void bl31_main(void)
  	PMF_CAPTURE_TIMESTAMP(bl_svc, BL31_EXIT, PMF_CACHE_MAINT);
  	console_flush();
  #endif