diff --git a/python/src/context.rs b/python/src/context.rs
index c5740a5bd..7e6ecb754 100644
--- a/python/src/context.rs
+++ b/python/src/context.rs
@@ -159,13 +159,13 @@ impl PySessionContext {
         &mut self,
         name: &str,
         path: &str,
-        table_partition_cols: Vec<(String, PyDataType)>,
+        table_partition_cols: Vec<(String, DataType)>,
         parquet_pruning: bool,
         file_extension: &str,
         py: Python,
     ) -> PyResult<()> {
         let mut options = ParquetReadOptions::default()
-            .table_partition_cols(convert_table_partition_cols(table_partition_cols))
+            .table_partition_cols(table_partition_cols)
             .parquet_pruning(parquet_pruning);
         options.file_extension = file_extension;
         let result = self.ctx.register_parquet(name, path, options);
@@ -254,13 +254,4 @@ impl PySessionContext {
     fn empty_table(&self) -> PyResult<PyDataFrame> {
         Ok(PyDataFrame::new(self.ctx.read_empty()?))
     }
-}
-
-fn convert_table_partition_cols(
-    table_partition_cols: Vec<(String, PyDataType)>,
-) -> Vec<(String, DataType)> {
-    table_partition_cols
-        .iter()
-        .map(|(name, t)| (name.clone(), t.data_type.clone()))
-        .collect()
 }
\ No newline at end of file