diff --git a/scripts/train_models.r b/scripts/train_models.r
index e4720cc79c8e07b7bb07af188f98d9de9304e376..56e59f220ac6c85f9c1558335f2769a4aed9966e 100644
--- a/scripts/train_models.r
+++ b/scripts/train_models.r
@@ -8,9 +8,6 @@ library("randomForest")
 # load in data
 load(file = "~/data.rdata")
 
-# remove RTN level
-data <- data %>% select(-c(RTN_LEVEL))
-
 # use index to partition data for training (90%) & validation
 training.index <- createDataPartition(data$RTN, p = 0.90, list = FALSE)