Fix tests
Merge branch 'patch-session'