diff --git a/modules/module-sh.el b/modules/module-sh.el
index 691cd06b6..7038ca99c 100644
--- a/modules/module-sh.el
+++ b/modules/module-sh.el
@@ -5,7 +5,8 @@
   :config
   (setq company-shell-delete-duplicates t))
 
-(associate! sh-mode :match "/\\.?z\\(sh/.*\\|profile\\|login\\|logout\\|shrc\\|shenv\\)$")
+(associate! sh-mode :match "\\.\\(ba\\|z\\)sh$")
+(associate! sh-mode :match "/\\.?z\\(sh\\(/.*\\|$\\)\\|profile\\|login\\|logout\\|shrc\\|shenv\\)$")
 (associate! sh-mode :match "/\\.?bash\\(/.*\\|rc\\|_profile\\)$")
 (after! sh-script
   (add-hook 'sh-mode-hook 'flycheck-mode)