Fixed issue with session controller