fix tests
make sure to use session check