diff --git a/pyvex/expr.py b/pyvex/expr.py
index ff2f07e6..5e2188dc 100644
--- a/pyvex/expr.py
+++ b/pyvex/expr.py
@@ -2,7 +2,7 @@
 import re
 from typing import List, Optional
 
-from archinfo import RegisterOffset, TmpVar
+from archinfo.types import RegisterOffset, TmpVar
 
 from .const import U8, U16, U32, U64, IRConst, get_type_size
 from .enums import IRCallee, IRRegArray, VEXObject, get_enum_from_int, get_int_from_enum
diff --git a/pyvex/stmt.py b/pyvex/stmt.py
index 21242e26..bb9cbf5d 100644
--- a/pyvex/stmt.py
+++ b/pyvex/stmt.py
@@ -1,7 +1,7 @@
 import logging
 from typing import Iterator, Optional
 
-from archinfo import RegisterOffset, TmpVar
+from archinfo.types import RegisterOffset, TmpVar
 
 from . import expr
 from .const import IRConst