diff --git a/opacus/utils/batch_memory_manager.py b/opacus/utils/batch_memory_manager.py
index 8f757e5d..c5d6dcc0 100644
--- a/opacus/utils/batch_memory_manager.py
+++ b/opacus/utils/batch_memory_manager.py
@@ -116,7 +116,7 @@ def wrap_data_loader(
     )
 
 
-class BatchMemoryManager(object):
+class BatchMemoryManager:
     """
     Context manager to manage memory consumption during training.