From 6494908e77b95832e4f59b2af2733e4fe5d37cfc Mon Sep 17 00:00:00 2001
From: Nao Pross <np@0hm.ch>
Date: Tue, 7 May 2024 02:56:30 +0200
Subject: Remove old comment

---
 polymatrix/expressionstate/mixins.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/polymatrix/expressionstate/mixins.py b/polymatrix/expressionstate/mixins.py
index 71add07..36dc5e7 100644
--- a/polymatrix/expressionstate/mixins.py
+++ b/polymatrix/expressionstate/mixins.py
@@ -35,7 +35,6 @@ class ExpressionStateMixin(
 
     def index(self, var: Variable) -> tuple[ExpressionStateMixin, IndexRange]:
         """ Index a variable and get its index range. """
-        # Unwrap if wrapped in expression object
         if not isinstance(var, Variable):
             raise ValueError("State can only index object of type Variable!")
 
@@ -50,7 +49,7 @@ class ExpressionStateMixin(
         return dataclasses.replace(
             self,
             n_variables=self.n_variables + size,
-            indices={**self.indices, var: index}
+            indices=self.indices | {var: index}
         ), index
 
     def register(self, var: Variable) -> ExpressionStateMixin:
-- 
cgit v1.2.1