Fix all

databricks · Aug 19, 2019 · f361f81 · f361f81
1 parent 5c065c9
commit f361f81
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/databricks/koalas/indexing.py b/databricks/koalas/indexing.py
@@ -450,7 +450,8 @@ def raiseNotImplemented(description):
             index_columns = self._kdf._internal.index_columns
             data_columns = [column for column in sdf.columns if column not in index_columns]
             internal = _InternalFrame(
-                sdf=sdf, data_columns=data_columns, index_map=self._kdf._internal.index_map)
+                sdf=sdf, data_columns=data_columns,
+                index_map=self._kdf._internal.index_map, column_index=column_index)
             kdf = DataFrame(internal)
         except AnalysisException:
             raise KeyError('[{}] don\'t exist in columns'