Add support for session rollbacks