diff --git a/lisa/trace.py b/lisa/trace.py
index b0b5a3cec87467a097ee6d14fbb9e0038c60fd99..5f0f42a5f260de5a869db5cfd588ddb899734b00 100644
--- a/lisa/trace.py
+++ b/lisa/trace.py
@@ -4328,7 +4328,7 @@ class _TraceCache(Loggable):
                 try:
                     data.sink_parquet(path, **kwargs)
                 # Some LazyFrame cannot be sunk lazily to a parquet file
-                except pl.InvalidOperationError:
+                except polars.exceptions.InvalidOperationError:
                     data.collect().write_parquet(path, **kwargs)
         else:
             data.to_parquet(path, **kwargs)