Fix session problem