Remove gradient cuz it was driving me crazy