update test logic