diff --git a/src/syntax/src/file_types.zig b/src/syntax/src/file_types.zig
index 13dc274..43795bf 100644
--- a/src/syntax/src/file_types.zig
+++ b/src/syntax/src/file_types.zig
@@ -343,7 +343,7 @@ pub const purescript = .{
 pub const python = .{
     .color = 0xffd845,
     .icon = "󰌠",
-    .extensions = .{"py"},
+    .extensions = .{"py", "pyi"},
     .comment = "#",
     .first_line_matches = .{ .prefix = "#!", .content = "python" },
     .language_server = .{"pylsp"},