Fix for updating math