diff --git a/.ci-support/fix-code-for-docs.sh b/.ci-support/fix-code-for-docs.sh
index b5a2809d3294f173d8e1bac914a0176b838a3d2c..7d6da022e787ef34d092ff9cba5a74e4cf3b5915 100755
--- a/.ci-support/fix-code-for-docs.sh
+++ b/.ci-support/fix-code-for-docs.sh
@@ -1,2 +1,2 @@
 #! /bin/bash
-sed -i "s/Dict\[str, DataInterface\]/Dict/" pytato/codegen/__init__.py
+sed -i "s/Dict\[str, DataInterface\]/Dict/" pytato/codegen.py