fix grad error