Gradient fix
Merge pull request #2484 from guardian/gradient-fix

Gradient fix