Fix for num gradients per block
param tweak