diff --git a/.coveragerc b/.coveragerc
index 7e4acd5b39..f4447255d5 100644
--- a/.coveragerc
+++ b/.coveragerc
@@ -2,6 +2,7 @@
 
 [run]
 source = MaxText
+         maxtext
 branch = True
 omit =
     tests/*