diff --git a/benchmark/python/benchmarks/lif_benchmark.py b/benchmark/python/benchmarks/lif_benchmark.py
index bd5f0c1..3fe9784 100644
--- a/benchmark/python/benchmarks/lif_benchmark.py
+++ b/benchmark/python/benchmarks/lif_benchmark.py
@@ -1,6 +1,6 @@
 import torch
 import numpy as np
-from mpact.models.lif import Block
+from mpact.models.lif import LIFSumOfSq
 from mpact_benchmark.utils.benchmark_utils import benchmark, Backends
 
 
@@ -35,10 +35,10 @@
         for dtype in [np.float64]
     ]
 )
-def SNN() -> torch.nn.Module:
-    """Spiking Neural Network."""
-    return Block()
+def LifSumOfSq() -> torch.nn.Module:
+    """LIF feeding into sum of squares."""
+    return LIFSumOfSq()
 
 
 if __name__ == "__main__":
-    SNN()
+    LifSumOfSq()
diff --git a/python/mpact/models/lif.py b/python/mpact/models/lif.py
index 37bf94f..fcb5a55 100644
--- a/python/mpact/models/lif.py
+++ b/python/mpact/models/lif.py
@@ -46,9 +46,9 @@
         return out
 
 
-class Block(torch.nn.Module):
+class LIFSumOfSq(torch.nn.Module):
     def __init__(self):
-        super(Block, self).__init__()
+        super(LIFSumOfSq, self).__init__()
         self.spike = LIF()
         self.layer = tdLayer(sqSum)
 
diff --git a/test/python/lif.py b/test/python/lif.py
index b7c4288..7dc797a 100644
--- a/test/python/lif.py
+++ b/test/python/lif.py
@@ -4,9 +4,9 @@
 
 from mpact.mpactbackend import mpact_jit, mpact_jit_compile, mpact_jit_run
 
-from mpact.models.lif import Block
+from mpact.models.lif import LIFSumOfSq
 
-net = Block()
+net = LIFSumOfSq()
 
 # Get a random (but reproducible) input, so that a
 # general sparse tensor appears after LIF.