Fix test session helper