diff --git a/split-services-and-network.py b/split-services-and-network.py
index d322d74b..8924bd85 100755
--- a/split-services-and-network.py
+++ b/split-services-and-network.py
@@ -34,6 +34,8 @@ for service_name, service in stack.items():
         del service["links"]
     if "ports" in service:
         del service["ports"]
+    if service_name in ports:
+        service["ports"] = [ ports[service_name] ]
 
 yaml.safe_dump(stack, open(output_file, "w"))