diff --git a/.github/workflows/integration.yml b/.github/workflows/integration.yml
index 19cb54f..7a88da3 100644
--- a/.github/workflows/integration.yml
+++ b/.github/workflows/integration.yml
@@ -118,7 +118,7 @@ jobs:
           NAMESPACE: ""
         run: cd ../../ && docker compose up
       - name: Move
-        run: mv ../../output/*.py .
+        run: mv ../../output/*.py ./gen
       - name: Run
         run: python main.py
   rust:
diff --git a/integration/python/gen/.gitkeep b/integration/python/gen/.gitkeep
new file mode 100644
index 0000000..e69de29
diff --git a/integration/python/main.py b/integration/python/main.py
index 9bf02fd..a04e654 100644
--- a/integration/python/main.py
+++ b/integration/python/main.py
@@ -1,11 +1,18 @@
 import json
 from pathlib import Path
-from .news import News
+from gen.news import News
 
-input = Path('../input.json').read_text()
 
-news = News.model_validate_json(input)
+def main():
+  input = Path('../input.json').read_text()
 
-output = json.dumps(news)
+  news = News.model_validate_json(input)
+
+  output = news.model_dump_json()
+
+  Path('../output.json').write_text(output)
+
+
+if __name__ == '__main__':
+  main()
 
-Path('../output.json').write_text(output)