Adjust to new session method API