Updated test logic.