Changed og correction logic