diff --git a/templates/leetcode.j2 b/templates/leetcode.j2
index 4391838..776dd6d 100644
--- a/templates/leetcode.j2
+++ b/templates/leetcode.j2
@@ -1,3 +1,4 @@
+from typing import List
 import pytest