diff --git a/ape_vyper/__init__.py b/ape_vyper/__init__.py
index f3de0527..debd2dfe 100644
--- a/ape_vyper/__init__.py
+++ b/ape_vyper/__init__.py
@@ -10,4 +10,4 @@ def config_class():
 
 @plugins.register(plugins.CompilerPlugin)
 def register_compiler():
-    return (".vy",), VyperCompiler
+    return (".vy", ".vyi"), VyperCompiler