diff --git a/tests/test_ab.py b/tests/test_ab.py
index 27ebd06a..ced8ba85 100644
--- a/tests/test_ab.py
+++ b/tests/test_ab.py
@@ -1,3 +1,4 @@
+import random
 import pytest
 from reasoning_gym.algorithmic.ab import ABConfig, ABDataset, compute_steps, generate_program