fix math mistake