diff --git a/bsb_neuron/__init__.py b/bsb_neuron/__init__.py
index 05f6b6e..bc39843 100644
--- a/bsb_neuron/__init__.py
+++ b/bsb_neuron/__init__.py
@@ -8,5 +8,5 @@
 from .adapter import NeuronAdapter
 from .simulation import NeuronSimulation
 
-__version__ = "4.0.1"
+__version__ = "4.0.2"
 __plugin__ = SimulationBackendPlugin(Simulation=NeuronSimulation, Adapter=NeuronAdapter)
diff --git a/bsb_neuron/adapter.py b/bsb_neuron/adapter.py
index c97ff91..2a20013 100644
--- a/bsb_neuron/adapter.py
+++ b/bsb_neuron/adapter.py
@@ -247,11 +247,11 @@ def __init__(self, model: "NeuronCell", instances: list):
     def __getitem__(self, item):
         # Boolean masking, kind of
         if getattr(item, "dtype", None) == bool or _all_bools(item):
-            return NeuronPopulation([p for p, b in zip(self, item) if b])
+            return NeuronPopulation(self._model, [p for p, b in zip(self, item) if b])
         elif getattr(item, "dtype", None) == int or _all_ints(item):
             if getattr(item, "ndim", None) == 0:
                 return super().__getitem__(item)
-            return NeuronPopulation([self[i] for i in item])
+            return NeuronPopulation(self._model, [self[i] for i in item])
         else:
             return super().__getitem__(item)
 
diff --git a/pyproject.toml b/pyproject.toml
index 7fc092f..af07b05 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -38,7 +38,7 @@ line-length = 90
 profile = "black"
 
 [tool.bumpversion]
-current_version = "4.0.1"
+current_version = "4.0.2"
 parse = "(?P<major>\\d+)\\.(?P<minor>\\d+)\\.(?P<patch>\\d+)"
 serialize = ["{major}.{minor}.{patch}"]
 search = "{current_version}"