Replace else branch with early exit