diff --git a/scripts/check_enterprise_imports.sh b/scripts/check_enterprise_imports.sh
index 4aacae53ad869..b78c8d1fa779f 100755
--- a/scripts/check_enterprise_imports.sh
+++ b/scripts/check_enterprise_imports.sh
@@ -9,7 +9,7 @@ source "$(dirname "${BASH_SOURCE[0]}")/lib.sh"
 cdroot
 
 set +e
-find . -regex ".*\.go" | grep -v "./enterprise" | grep -v "./scripts/auditdocgen/main.go" | xargs grep -n "github.com/coder/coder/enterprise"
+find . -regex ".*\.go" | grep -v "./enterprise" | grep -v ./scripts/auditdocgen/ --include="*.go" | xargs grep -n "github.com/coder/coder/enterprise"
 # reverse the exit code because we want this script to fail if grep finds anything.
 status=$?
 set -e