Update handler test with new session interface