Fix favor modifier check for assignment in condition

Previous code was checking in the body of the while/if/unless, too.