fix for diagonal gauss nll.