diff --git a/crates/polars/tests/it/lazy/expressions/arity.rs b/crates/polars/tests/it/lazy/expressions/arity.rs
index a4cfc7796a66..17d4a1470110 100644
--- a/crates/polars/tests/it/lazy/expressions/arity.rs
+++ b/crates/polars/tests/it/lazy/expressions/arity.rs
@@ -260,7 +260,7 @@ fn test_binary_over_3930() -> PolarsResult<()> {
     ]?;
 
     let ss = col("score").pow(2);
-    let mdiff = (ss.clone().shift(-1) - ss.shift(1)) / lit(2);
+    let mdiff = (ss.clone().shift(lit(-1)) - ss.shift(lit(1))) / lit(2);
     let out = df.lazy().select([mdiff.over([col("class")])]).collect()?;
 
     let out = out.column("score")?;
diff --git a/crates/polars/tests/it/lazy/projection_queries.rs b/crates/polars/tests/it/lazy/projection_queries.rs
index 4f963c8385d7..40722e4dad2b 100644
--- a/crates/polars/tests/it/lazy/projection_queries.rs
+++ b/crates/polars/tests/it/lazy/projection_queries.rs
@@ -136,7 +136,7 @@ fn test_projection_5086() -> PolarsResult<()> {
         .select([
             col("a"),
             col("b")
-                .xor(col("b").shift(1).over([col("a")]))
+                .xor(col("b").shift(lit(1)).over([col("a")]))
                 .fill_null(lit(true))
                 .alias("keep"),
         ])