diff --git a/application/src/main/data/upgrade/3.4.1/schema_update.sql b/application/src/main/data/upgrade/3.4.1/schema_update.sql
index 0f4354c0ca..8dceb0115d 100644
--- a/application/src/main/data/upgrade/3.4.1/schema_update.sql
+++ b/application/src/main/data/upgrade/3.4.1/schema_update.sql
@@ -14,6 +14,7 @@
 -- limitations under the License.
 --
 
+-- AUDIT LOGS MIGRATION START
 DO
 $$
     DECLARE table_partition RECORD;
@@ -73,8 +74,10 @@ BEGIN
     WHERE created_time >= start_time_ms AND created_time < end_time_ms;
 END;
 $$;
+-- AUDIT LOGS MIGRATION END
 
 
+-- EDGE EVENTS MIGRATION START
 DO
 $$
     DECLARE table_partition RECORD;
@@ -131,3 +134,4 @@ BEGIN
     WHERE created_time >= start_time_ms AND created_time < end_time_ms;
 END;
 $$;
+-- EDGE EVENTS MIGRATION END