Fix Yotsuba B gradient