WIKI-409 fix bug that missing gradient