diff --git a/schemas/api/trains/TrainDetails.json b/schemas/api/trains/TrainDetails.json
index 8c4046a..000bbf1 100644
--- a/schemas/api/trains/TrainDetails.json
+++ b/schemas/api/trains/TrainDetails.json
@@ -218,7 +218,7 @@
             }
         },
         "pis": {
-            "$ref": "../pis-object.json"
+            "$ref": "./schemas/api/pis-object.json"
         }
     }
 }
\ No newline at end of file
diff --git a/scripts/build.sh b/scripts/build.sh
index aa5a5f8..8c7ab90 100644
--- a/scripts/build.sh
+++ b/scripts/build.sh
@@ -18,7 +18,7 @@ for file in $FILES; do
     npx --yes json-schema-to-typescript "$file" > "../../../gen/ts/${clean_name}.ts"
     
     # Generate Go
-    go-jsonschema -p contracts "$file" > "gen/go/models/${clean_name}.go"
+    go-jsonschema -p contracts "$file" > "./gen/go/models/${clean_name}.go"
 done
 
 echo "✅ Generated single TS package in gen/ts"