diff --git a/forward_engineering/generateContainerScript.js b/forward_engineering/generateContainerScript.js
index 7cf61a0..c3106dc 100644
--- a/forward_engineering/generateContainerScript.js
+++ b/forward_engineering/generateContainerScript.js
@@ -60,7 +60,7 @@ const generateIndexes = indexesData => {
 };
 
 const generateEdge = (from, to, relationship, edgeData) => {
-	const edgeName = transformToValidGremlinName(relationship.name);
+	const edgeName = transformToValidGremlinName(relationship.code || relationship.name);
 	const propertiesScript = addPropertiesScript(relationship, edgeData);
 
 	return `${from}.addE(${JSON.stringify(edgeName)}).\n${DEFAULT_INDENT}to(${to})${propertiesScript}`;