diff --git a/sqlalchemy-stubs/orm/session.pyi b/sqlalchemy-stubs/orm/session.pyi
index af3cbed..d39b5d6 100644
--- a/sqlalchemy-stubs/orm/session.pyi
+++ b/sqlalchemy-stubs/orm/session.pyi
@@ -45,6 +45,8 @@ class Session(_SessionClassMethods):
                  binds: Optional[Any] = ..., extension: Optional[Any] = ...,
                  info: Optional[Any] = ..., query_cls: Any = ...) -> None: ...
     connection_callable: Any = ...
+    def __enter__(self): ...
+    def __exit__(self, type, value, traceback): ...
     @property
     def info(self): ...
     def begin(self, subtransactions: bool = ..., nested: bool = ...) -> SessionTransaction: ...