diff --git a/tests/trainer/test_trainer.py b/tests/trainer/test_trainer.py
index 9cdb02468b3034..f48265ffa58168 100644
--- a/tests/trainer/test_trainer.py
+++ b/tests/trainer/test_trainer.py
@@ -2398,7 +2398,7 @@ def test_bnb_adam8bit_no_bnb(self):
 
         # Pretend that bnb does not exist, even if installed. By setting bnb to None, importing
         # bnb will fail even if bnb is installed.
-        with patch.dict("sys.modules", {"bnb.optim": None}):
+        with patch.dict("sys.modules", {"bitsandbytes.optim": None}):
             with self.assertRaises(ValueError):
                 Trainer.get_optimizer_cls_and_kwargs(args)