Fix ugly ass gradient fo realz

Change-Id: Idd896c615355f2ae26f7e27509c48f4a4a181ed4