diff --git a/R/codegen.R b/R/codegen.R
index 1da3430..82f06b9 100644
--- a/R/codegen.R
+++ b/R/codegen.R
@@ -157,7 +157,7 @@ codegen.workflow <- function(tm_workflow, path = "scoreCode.R", rds = "model.rds
     
   if (!is.null(inputs)) {
     
-    if (!is.vector(names(hmeqTrain))) stop("inputs must be a vector names")
+    if (!is.vector(inputs)) stop("inputs must be a vector names")
     
     predictors <- inputs