Revert "Fix gradient. Credit to UBER."

This reverts commit fb819a00b6d6912009e2597ec000a97fc8dab2a2.