diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index 9b923bf53..4e500b424 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -32,15 +32,20 @@ jobs:
       - name: Setup environment
         uses: ./.github/actions/setup-environment
 
-      - name: Run ATS and Tests
-        uses: ./.github/actions/run-ats
-        timeout-minutes: 15
+      - name: Test with pytest
+        timeout-minutes: 5
+        run: |
+          uv run pytest \
+            -n auto \
+            --cov src \
+            --timeout 15 \
+            -o junit_suite_name="${{github.job}}" \
+            tests/unit
+
+      - uses: ./.github/actions/report
         with:
-          default_tests: "tests/unit"
-          codecov_static_token: ${{ secrets.CODECOV_STATIC_TOKEN }}
+          flag: unit-tests
           codecov_token: ${{ secrets.CODECOV_TOKEN }}
-          collect_args: "--timeout 15"
-          codecov_flags: unit-tests
 
   codemod-tests:
     needs: access-check
diff --git a/src/codegen/extensions/tools/create_file.py b/src/codegen/extensions/tools/create_file.py
index 3a54303ff..77d8dd20d 100644
--- a/src/codegen/extensions/tools/create_file.py
+++ b/src/codegen/extensions/tools/create_file.py
@@ -45,6 +45,7 @@ def create_file(codebase: Codebase, filepath: str, content: str) -> CreateFileOb
                 filepath=filepath,
                 content="",
                 line_count=0,
+                raw_content="",
             ),
         )
 
@@ -72,5 +73,6 @@ def create_file(codebase: Codebase, filepath: str, content: str) -> CreateFileOb
                 filepath=filepath,
                 content="",
                 line_count=0,
+                raw_content="",
             ),
         )
diff --git a/src/codegen/extensions/tools/view_file.py b/src/codegen/extensions/tools/view_file.py
index 21fa347a6..fbfcd8b02 100644
--- a/src/codegen/extensions/tools/view_file.py
+++ b/src/codegen/extensions/tools/view_file.py
@@ -133,6 +133,7 @@ def view_file(
 Ensure that this is indeed the correct filepath, else keep searching to find the correct fullpath.""",
             filepath=filepath,
             content="",
+            raw_content="",
             line_count=0,
             start_line=start_line,
             end_line=end_line,