fix session update in tests