Fixed tests to work with new session controller