diff --git a/DESCRIPTION b/DESCRIPTION
index 139ea46..2bf1388 100644
--- a/DESCRIPTION
+++ b/DESCRIPTION
@@ -1,7 +1,7 @@
 Package: jskm
 Title: Kaplan-Meier Plot with 'ggplot2'
-Version: 0.5
-Date: 2023-03-18
+Version: 0.5.1
+Date: 2023-07-18
 Authors@R: c(person("Jinseob", "Kim", email = "jinseob2kim@gmail.com", role = c("aut", "cre"),  comment = c(ORCID = "0000-0002-9403-605X")),
              person("Zarathu", role = c("cph", "fnd"))
              )
diff --git a/NEWS.md b/NEWS.md
index 950fd9a..6827317 100644
--- a/NEWS.md
+++ b/NEWS.md
@@ -1,3 +1,7 @@
+# jskm 0.5.1
+
+* Fix: line color problem when apply landmark analysis
+
 # jskm 0.5
 
 * Add competing risk analysis to `jskm`
diff --git a/R/jskm.R b/R/jskm.R
index 7730c65..16bd4ce 100644
--- a/R/jskm.R
+++ b/R/jskm.R
@@ -250,7 +250,7 @@ jskm <- function(sfit,
       }
       
       df11 <- rbind(subset(df, time < cut.landmark), df2[, names(df)]) 
-      df <- rbind(df11, data.frame(time = cut.landmark, n.risk = summary(sfit, times = cut.landmark)$n.risk[[1]],  n.event = 0, n.censor = 0, surv = 1, strata = factor(ystratalabs, levels = levels(df$strata)), upper = 1, lower = 1))
+      df <- rbind(df11, data.frame(time = cut.landmark, n.risk = summary(sfit, times = cut.landmark)$n.risk[[1]],  n.event = 0, n.censor = 0, surv = 1, strata = levels(df$strata), upper = 1, lower = 1))
     } else{
       if (is.null(status.cmprsk)){
         status.cmprsk <- sfit$states[2]
@@ -268,7 +268,7 @@ jskm <- function(sfit,
                      by = c("time", "strata"))
       }
       df11 <- rbind(subset(df, time < cut.landmark), df2[, names(df)]) 
-      df <- rbind(df11, data.frame(time = cut.landmark, n.risk = summary(sfit, times = cut.landmark)$n.risk[[1]],  n.event = 0, n.censor = 0, surv = 0, strata = factor(ystratalabs, levels = levels(df$strata)), upper = 0, lower = 0))
+      df <- rbind(df11, data.frame(time = cut.landmark, n.risk = summary(sfit, times = cut.landmark)$n.risk[[1]],  n.event = 0, n.censor = 0, surv = 0, strata = levels(df$strata), upper = 0, lower = 0))
     }  
     
     
@@ -353,7 +353,7 @@ jskm <- function(sfit,
   } else{
     p <- p +
       scale_linetype_manual(name = ystrataname, values=linetype) +  
-      geom_step(data = subset(df, time < cut.landmark), linewidth = 0.75) + geom_step(data = subset(df, time >= cut.landmark), linewidth = 0.75) + 
+      geom_step(data = subset(df, time >= cut.landmark), linewidth = 0.75) + geom_step(data = subset(df, time < cut.landmark), linewidth = 0.75) + 
       scale_colour_brewer(name = ystrataname, palette=linecols)
   }