And yet again fix gradient