diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index b178573..ba45573 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -4,7 +4,7 @@ repos:
     rev: v0.2.1
     hooks:
       - id: ruff
-        args: [ --fix, --select, I ]
+        args: [ --fix, --extend-select, I ]
       - id: ruff-format
 
   - repo: https://github.com/codespell-project/codespell
diff --git a/live_bargaining/tests.py b/live_bargaining/tests.py
index 18e6746..de42664 100644
--- a/live_bargaining/tests.py
+++ b/live_bargaining/tests.py
@@ -268,7 +268,6 @@ def play_round(self):
 
         Y = list(self.session.config["prod_fct"].values())[1]
 
-        num_real_rounds = 4
         expected_payoffs = {
             1: [0, 0, 0],
             2: [50, 25, 25],
diff --git a/survey/__init__.py b/survey/__init__.py
index d36a294..313aa3e 100644
--- a/survey/__init__.py
+++ b/survey/__init__.py
@@ -7,7 +7,6 @@
     BasePlayer,
     BaseSubsession,
     Page,
-    cu,
     models,
     widgets,
 )