diff --git a/src/python/nimbusml/internal/utils/data_schema.py b/src/python/nimbusml/internal/utils/data_schema.py
index e99f65f4..60717bda 100644
--- a/src/python/nimbusml/internal/utils/data_schema.py
+++ b/src/python/nimbusml/internal/utils/data_schema.py
@@ -610,15 +610,8 @@ def read_schema_file(
                 df = read_csv(filepath_or_buffer, nrows=nrows, **pd_options)
 
             # We remove integers as column names.
-            if sys.version_info < (3, 0):
-                df.columns = \
-                    ['c' + str(_) if
-                        (not isinstance(_, str) and not isinstance(_, unicode))
-                        else _ for _ in df.columns]
-            else:
-                df.columns = [
-                    'c' + str(_) if not isinstance(_, str)
-                    else _ for _ in df.columns]
+            df.columns = [_ if isinstance(_, six.string_types)
+                          else 'c' + str(_) for _ in df.columns]
 
             if isinstance(pd_options.get('dtype', None), dict):
                 # We overwrite types if specified.