diff --git a/pwndbg/heap/ptmalloc.py b/pwndbg/heap/ptmalloc.py
index 6c640e5c9..0158b47a3 100644
--- a/pwndbg/heap/ptmalloc.py
+++ b/pwndbg/heap/ptmalloc.py
@@ -5,6 +5,7 @@ from enum import Enum
 from typing import Any
 from typing import Dict
 from typing import Set
+from typing import Union
 
 import gdb
 
diff --git a/pwndbg/lib/memoize.py b/pwndbg/lib/memoize.py
index 7e2a1d740..52b8155da 100644
--- a/pwndbg/lib/memoize.py
+++ b/pwndbg/lib/memoize.py
@@ -10,6 +10,7 @@ from collections.abc import Hashable
 from typing import Any
 from typing import Callable
 from typing import Dict
+from typing import List
 
 debug = False