diff --git a/python_models8/connectors/my_connector.py b/python_models8/connectors/my_connector.py
index 4c28e0f..87af0bb 100644
--- a/python_models8/connectors/my_connector.py
+++ b/python_models8/connectors/my_connector.py
@@ -3,7 +3,7 @@
 from spinn_utilities.overrides import overrides
 from pacman.model.graphs.common import Slice
 from spynnaker.pyNN.models.neural_projections import SynapseInformation
-from spynnaker.pyNN.types import WEIGHTS
+from spynnaker.pyNN.types import Weights
 from spynnaker.pyNN.models.neural_projections.connectors import (
     AbstractConnector, AbstractGenerateConnectorOnHost)
 
@@ -51,7 +51,7 @@ def create_synaptic_block(
         raise NotImplementedError
 
     @overrides(AbstractConnector.get_weight_variance)
-    def get_weight_variance(self, weights: WEIGHTS,
+    def get_weight_variance(self, weights: Weights,
                             synapse_info: SynapseInformation) -> float:
         # TODO: update accordingly
         raise NotImplementedError