diff --git a/benchmarks/benchmark_kyber.py b/benchmarks/benchmark_kyber.py
index 6ccd0a0..ddb9c92 100644
--- a/benchmarks/benchmark_kyber.py
+++ b/benchmarks/benchmark_kyber.py
@@ -45,7 +45,7 @@ def benchmark_kyber(Kyber, name, count):
         enc_times.append(time() - t1)
 
         t2 = time()
-        dec = Kyber.decaps(sk, c)
+        _ = Kyber.decaps(sk, c)
         dec_times.append(time() - t2)
 
     avg_keygen = sum(keygen_times) / count
diff --git a/pyproject.toml b/pyproject.toml
index a8f43fe..bef2ebf 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,2 +1,6 @@
 [tool.black]
 line-length = 79
+
+[tool.ruff.lint]
+# Avoid enforcing Ambiguous variable name (`E741`)
+ignore = ["E741"]
diff --git a/src/kyber_py/kyber/__init__.py b/src/kyber_py/kyber/__init__.py
index 59d07b8..ab15900 100644
--- a/src/kyber_py/kyber/__init__.py
+++ b/src/kyber_py/kyber/__init__.py
@@ -1 +1,3 @@
 from .default_parameters import Kyber512, Kyber768, Kyber1024
+
+__all__ = [Kyber512, Kyber768, Kyber1024]
diff --git a/src/kyber_py/ml_kem/__init__.py b/src/kyber_py/ml_kem/__init__.py
index 2b62e3f..7d05898 100644
--- a/src/kyber_py/ml_kem/__init__.py
+++ b/src/kyber_py/ml_kem/__init__.py
@@ -1 +1,3 @@
 from .default_parameters import ML_KEM_512, ML_KEM_768, ML_KEM_1024
+
+__all__ = [ML_KEM_512, ML_KEM_768, ML_KEM_1024]
diff --git a/tests/test_polynomial.py b/tests/test_polynomial.py
index 6dc8796..518c6c1 100644
--- a/tests/test_polynomial.py
+++ b/tests/test_polynomial.py
@@ -1,5 +1,4 @@
 import unittest
-from random import randint
 from kyber_py.polynomials.polynomials import PolynomialRingKyber