diff --git a/nngeometry/generator/jacobian/grads.py b/nngeometry/generator/jacobian/grads.py
index e01e481..d283e12 100644
--- a/nngeometry/generator/jacobian/grads.py
+++ b/nngeometry/generator/jacobian/grads.py
@@ -155,7 +155,6 @@ def kfac_xx(cls, buffer, mod, layer, x, gy):
 
     @classmethod
     def kfac_gg(cls, buffer, mod, layer, x, gy):
-        spatial_locations = gy.size(2) * gy.size(3)
         os = gy.size(1)
         # DS_tilda in KFC
         DS_tilda = gy.permute(0, 2, 3, 1).contiguous().view(-1, os)