diff --git a/python/sparknlp/annotator/tf_ner_dl_graph_builder.py b/python/sparknlp/annotator/tf_ner_dl_graph_builder.py
index fc3abed948c3ea..5ffc598d2598a2 100644
--- a/python/sparknlp/annotator/tf_ner_dl_graph_builder.py
+++ b/python/sparknlp/annotator/tf_ner_dl_graph_builder.py
@@ -66,7 +66,7 @@ def setInputCols(self, *value):
 
         Parameters
         ----------
-        *value : str
+        *value : List[str]
             Input columns for the annotator
         """
         if type(value[0]) == str or type(value[0]) == list:
diff --git a/python/sparknlp/base/finisher.py b/python/sparknlp/base/finisher.py
index 8b16d5d3dd46ff..6f82ae2191a5d4 100644
--- a/python/sparknlp/base/finisher.py
+++ b/python/sparknlp/base/finisher.py
@@ -123,7 +123,7 @@ def setInputCols(self, *value):
 
         Parameters
         ----------
-        *value : str
+        *value : List[str]
             Input columns for the annotator
         """
         if len(value) == 1 and type(value[0]) == list:
diff --git a/python/sparknlp/base/multi_document_assembler.py b/python/sparknlp/base/multi_document_assembler.py
index 46fdae2f4010db..3415ff09e08eb9 100644
--- a/python/sparknlp/base/multi_document_assembler.py
+++ b/python/sparknlp/base/multi_document_assembler.py
@@ -104,7 +104,7 @@ def setInputCols(self, *value):
 
         Parameters
         ----------
-        *value : str
+        *value : List[str]
             Input columns for the annotator
         """
         if len(value) == 1 and type(value[0]) == list: