Fix accidental logic fail