diff --git a/src/lte/doc/source/figures/nas-activate-dedicated-bearer.seqdiag b/src/lte/doc/source/figures/nas-activate-dedicated-bearer.seqdiag
index 87c3073c8..c4b5d62d7 100644
--- a/src/lte/doc/source/figures/nas-activate-dedicated-bearer.seqdiag
+++ b/src/lte/doc/source/figures/nas-activate-dedicated-bearer.seqdiag
@@ -5,12 +5,12 @@ diagram {
 	LteHelper => LteUeNetDevice [label="ActivateDedicatedEpsBearer(Tft)"] {
 		LteUeNetDevice => EpcUeNas [label="ActivateDedicatedEpsBearer(Tft)"] {
 			EpcUeNas => EpcHelper [label="ActivateEpsBearer(UeDevice, EnbDevice, Tft)"] {
-					EpcHelper => EpcSgwPgwApplication [label="ActivateS1Bearer (UE IP, eNB IP, Tft)", return="TEID"] {
+					EpcHelper => EpcSgwPgwApplication [label="ActivateS1Bearer (UE IP, eNB IP, IMSI, Tft)", return="TEID"] {
 						EpcSgwPgwApplication => EpcSgwPgwApplication [label="Store UE IP<->eNB IP mapping"];
 						EpcSgwPgwApplication => EpcSgwPgwApplication [label="Create GTP-U tunnel endpoint"];
 					}
-					EpcHelper => EpcEnbApplication [label="ErabSetupRequest(TEID, RNTI)"] {
-						EpcEnbApplication => LteEnbRrc [label="SetupRadioBearer", return="LCID"] {
+					EpcHelper => EpcEnbApplication [label="ErabSetupRequest(TEID, IMSI)"] {
+						EpcEnbApplication => LteEnbRrc [label="RadioBearerSetupRequest (IMSI)", return="RadioBearerSetupCompleted (RNTI, LCID)"] {
 							LteEnbRrc => LteUeRrc [label="RRC connection reconfiguration"];
 						}
 						EpcEnbApplication -> EpcEnbApplication [label="Create GTP-U tunnel endpoint (TEID)"];
@@ -21,4 +21,4 @@ diagram {
 		}
 	}
 }
-		
+
diff --git a/src/lte/doc/source/figures/nas-attach.seqdiag b/src/lte/doc/source/figures/nas-attach.seqdiag
index 1220a2bac..5cf99fca7 100644
--- a/src/lte/doc/source/figures/nas-attach.seqdiag
+++ b/src/lte/doc/source/figures/nas-attach.seqdiag
@@ -3,24 +3,28 @@ diagram {
 	LteHelper; LteUeNetDevice; EpcUeNas; LteUeRrc; LteEnbRrc; EpcEnbApplication; EpcSgwPgwApplication; EpcHelper;	
 	
 	LteHelper -> LteUeNetDevice [label="Attach (eNB)"];
-	LteHelper -> LteUeRrc [label="Set Cell ID"];
+	LteHelper -> LteUeRrc [label="Set eNB"];
 	LteUeNetDevice -> EpcUeNas [label="Connect"];
-	EpcUeNas -> LteUeRrc [label="Connect"];
-	
+	EpcUeNas => LteUeRrc [label="Connect"] {
+		LteUeRrc => LteEnbRrc [label="Connection Request (IMSI)"] {
+			=== RRC connection establishment ===
+			LteEnbRrc -> LteEnbRrc [label="store IMSI<->RNTI mapping"];			
+		}
+	}	
 	
 	EpcUeNas => EpcHelper [label="ActivateEpsBearer(UeDevice, EnbDevice, Tft::Default)"] {
-			EpcHelper => EpcSgwPgwApplication [label="ActivateS1Bearer (UE IP, eNB IP, Tft::Default)", return="TEID"] {
-				EpcSgwPgwApplication => EpcSgwPgwApplication [label="Store UE IP<->eNB IP mapping"];
-				EpcSgwPgwApplication => EpcSgwPgwApplication [label="Create GTP-U tunnel endpoint"];
-			}
-			EpcHelper => EpcEnbApplication [label="ErabSetupRequest(TEID, RNTI)"] {
-				EpcEnbApplication => LteEnbRrc [label="SetupRadioBearer", return="LCID"] {
-					LteEnbRrc => LteUeRrc [label="RRC connection reconfiguration"];
-				}
-				EpcEnbApplication -> EpcEnbApplication [label="Create GTP-U tunnel endpoint (TEID)"];
-				EpcEnbApplication -> EpcEnbApplication [label="store TEID<->(RNTI,LCID) mapping"];
-			}  
+		EpcHelper => EpcSgwPgwApplication [label="ActivateS1Bearer (UE IP, eNB IP, IMSI, Tft::Default)", return="TEID"] {
+			EpcSgwPgwApplication => EpcSgwPgwApplication [label="Store UE IP<->eNB IP mapping"];
+			EpcSgwPgwApplication => EpcSgwPgwApplication [label="Create GTP-U tunnel endpoint"];
 		}
+		EpcHelper => EpcEnbApplication [label="ErabSetupRequest(TEID, IMSI)"] {
+			EpcEnbApplication => LteEnbRrc [label="RadioBearerSetupRequest (IMSI)", return="RadioBearerSetupCompleted (RNTI, LCID)"] {
+				LteEnbRrc => LteUeRrc [label="RRC connection reconfiguration"];
+			}
+			EpcEnbApplication -> EpcEnbApplication [label="Create GTP-U tunnel endpoint (TEID)"];
+			EpcEnbApplication -> EpcEnbApplication [label="store TEID<->(RNTI,LCID) mapping"];
+		}  
+	}
 	
 }