1 files changed, 4 insertions, 4 deletions
diff --git a/pyecsca/ec/op.py b/pyecsca/ec/op.py
index 43a84c0..72b06f3 100644
--- a/pyecsca/ec/op.py
+++ b/pyecsca/ec/op.py
@@ -1,6 +1,6 @@
-from ast import Module, walk, Name, BinOp, Constant, operator, Mult, Div, Add, Sub, Pow
+from ast import Module, walk, Name, BinOp, Constant, operator, Mult, Div, Add, Sub, Pow, Assign
 from types import CodeType
-from typing import FrozenSet, Optional
+from typing import FrozenSet, Optional, cast
 
 from .mod import Mod
 
@@ -15,8 +15,8 @@ class CodeOp(object):
 
     def __init__(self, code: Module):
         self.code = code
-        assign = code.body[0]
-        self.result = assign.targets[0].id
+        assign = cast(Assign, code.body[0])
+        self.result = cast(Name, assign.targets[0]).id
         params = set()
         variables = set()
         constants = set()