Fix session tests