Add == to assignment in condition