diff --git a/.github/workflows/codegen.yml b/.github/workflows/codegen.yml
index 87d65c48..13f5773d 100644
--- a/.github/workflows/codegen.yml
+++ b/.github/workflows/codegen.yml
@@ -85,7 +85,7 @@ jobs:
           ERLANG_OUTPUT_PATH: ../../src
         run: |
           # Remove any previously generated files
-          find . -type f -exec grep -l "DO NOT EDIT, AUTO-GENERATED CODE" {} \; | xargs rm -f
+          find src -type f -name "*.erl" -exec grep -l "DO NOT EDIT, AUTO-GENERATED CODE" {} \; | xargs rm -f
 
           # Jump to the codegen
           cd tmp/aws-codegen