diff --git a/crates/polars-plan/src/dsl/function_expr/shrink_type.rs b/crates/polars-plan/src/dsl/function_expr/shrink_type.rs
index dbf29308e191..0dee54f79e8f 100644
--- a/crates/polars-plan/src/dsl/function_expr/shrink_type.rs
+++ b/crates/polars-plan/src/dsl/function_expr/shrink_type.rs
@@ -1,35 +1,38 @@
 use super::*;
 
 pub(super) fn shrink(s: Series) -> PolarsResult<Series> {
-    if s.dtype().is_numeric() {
-        if s.dtype().is_float() {
-            s.cast(&DataType::Float32)
-        } else if s.dtype().is_unsigned_integer() {
-            let max = s.max_reduce()?.value().extract::<u64>().unwrap_or(0_u64);
-            if cfg!(feature = "dtype-u8") && max <= u8::MAX as u64 {
-                s.cast(&DataType::UInt8)
-            } else if cfg!(feature = "dtype-u16") && max <= u16::MAX as u64 {
-                s.cast(&DataType::UInt16)
-            } else if max <= u32::MAX as u64 {
-                s.cast(&DataType::UInt32)
-            } else {
-                Ok(s)
-            }
-        } else {
-            let min = s.min_reduce()?.value().extract::<i64>().unwrap_or(0_i64);
-            let max = s.max_reduce()?.value().extract::<i64>().unwrap_or(0_i64);
+    if !s.dtype().is_numeric() {
+        return Ok(s);
+    }
+
+    if s.dtype().is_float() {
+        return s.cast(&DataType::Float32);
+    }
+
+    if s.dtype().is_unsigned_integer() {
+        let max = s.max_reduce()?.value().extract::<u64>().unwrap_or(0_u64);
 
-            if cfg!(feature = "dtype-i8") && min >= i8::MIN as i64 && max <= i8::MAX as i64 {
-                s.cast(&DataType::Int8)
-            } else if cfg!(feature = "dtype-i16") && min >= i16::MIN as i64 && max <= i16::MAX as i64 {
-                s.cast(&DataType::Int16)
-            } else if min >= i32::MIN as i64 && max <= i32::MAX as i64 {
-                s.cast(&DataType::Int32)
-            } else {
-                Ok(s)
-            }
+        if cfg!(feature = "dtype-u8") && max <= u8::MAX as u64 {
+            s.cast(&DataType::UInt8)
+        } else if cfg!(feature = "dtype-u16") && max <= u16::MAX as u64 {
+            s.cast(&DataType::UInt16)
+        } else if max <= u32::MAX as u64 {
+            s.cast(&DataType::UInt32)
+        } else {
+            Ok(s)
         }
     } else {
-        Ok(s)
+        let min = s.min_reduce()?.value().extract::<i64>().unwrap_or(0_i64);
+        let max = s.max_reduce()?.value().extract::<i64>().unwrap_or(0_i64);
+
+        if cfg!(feature = "dtype-i8") && min >= i8::MIN as i64 && max <= i8::MAX as i64 {
+            s.cast(&DataType::Int8)
+        } else if cfg!(feature = "dtype-i16") && min >= i16::MIN as i64 && max <= i16::MAX as i64 {
+            s.cast(&DataType::Int16)
+        } else if min >= i32::MIN as i64 && max <= i32::MAX as i64 {
+            s.cast(&DataType::Int32)
+        } else {
+            Ok(s)
+        }
     }
 }