Math fix

Math fix