Fix test logic.