diff --git a/.env.development b/.env.development
index e80e73b..cd8d8ea 100644
--- a/.env.development
+++ b/.env.development
@@ -1,3 +1,3 @@
 VITE_HASURA_GRAPHQL_ENDPOINT = "https://test.hasura.magicsk.eu/v1/graphql"
 VITE_HASURA_GRAPHQL_SECRET = "c2akHldMLELW7z5RFUMu8AnUxyHAXAThFzW5JZK7HejgMjG5cn7kr1m0C0Vefdbc"
-VITE_INJECT_MODE = "jamstack"
\ No newline at end of file
+VITE_INJECT_MODE = "devServer"
\ No newline at end of file
diff --git a/codegen.yml b/codegen.yml
index d1d3c96..77b770c 100644
--- a/codegen.yml
+++ b/codegen.yml
@@ -1,6 +1,6 @@
 overwrite: true
 schema: ${VITE_HASURA_GRAPHQL_ENDPOINT}
-documents: './src/pages/**/*.graphql'
+documents: './src/**/*.graphql'
 generates:
   src/generated/graphql.ts:
     plugins:
diff --git a/vite.config.ts b/vite.config.ts
index 8b84d1a..5cc1f52 100644
--- a/vite.config.ts
+++ b/vite.config.ts
@@ -26,6 +26,7 @@ export default ({ command, mode }) => {
         graphQLSecret: env.VITE_HASURA_GRAPHQL_SECRET,
         cwd: process.cwd(),
         injectLowcode: true,
+        injectMode: env.VITE_NETLIFY ? "jamstack" : env.VITE_INJECT_MODE,
         features: {
           tippy: true,
           generator: true,