diff --git a/torchcache/torchcache.py b/torchcache/torchcache.py
index 81a86a4..3c38f1a 100644
--- a/torchcache/torchcache.py
+++ b/torchcache/torchcache.py
@@ -548,8 +548,8 @@ def _cache_to_file(self, embedding: Tensor, hash_val: int) -> None:
         if self.zstd_compression:
             raw_data = zstd.compress(
                 raw_data,
-                level=self.zstd_compression_level,
-                threads=self.zstd_compression_threads,
+                self.zstd_compression_level,
+                self.zstd_compression_threads,
             )
             logger.debug(f"Compressed data size: {len(raw_data)}")