added back session tests, hope very thing work proper~