Fix stupid bug with conditional