diff --git a/plugins/modules/podman_container.py b/plugins/modules/podman_container.py
index 071952e..f88ffb2 100644
--- a/plugins/modules/podman_container.py
+++ b/plugins/modules/podman_container.py
@@ -1627,20 +1627,12 @@ class PodmanContainerDiff:
 
     # TODO(sshnaidm) Need to add port ranges support
     def diffparam_publish(self):
-        ports = self.info['networksettings']['ports']
-        # before 2.0.0 we have ports as a list, after 2.0.0 it's a dictionary
-        if isinstance(ports, list):
-            before = [":".join([
-                i['hostip'],
-                str(i["hostport"]),
-                str(i["containerport"])
-            ]).strip(':') for i in ports]
-        elif isinstance(ports, dict):
-            before = [":".join([
-                j[0]['hostip'],
-                str(j[0]["hostport"]),
-                i.replace('/tcp', '')
-            ]).strip(':') for i, j in ports.items()]
+        ports = self.info['hostconfig']['portbindings']
+        before = [":".join([
+            j[0]['hostip'],
+            str(j[0]["hostport"]),
+            i.replace('/tcp', '')
+        ]).strip(':') for i, j in ports.items()]
         after = self.params['publish'] or []
         if self.params['publish_all']:
             image_ports = self.image_info['config'].get('exposedports', {})
diff --git a/tests/integration/targets/podman_container_idempotency/tasks/idem_ports.yml b/tests/integration/targets/podman_container_idempotency/tasks/idem_ports.yml
index c039b58..9e029b8 100644
--- a/tests/integration/targets/podman_container_idempotency/tasks/idem_ports.yml
+++ b/tests/integration/targets/podman_container_idempotency/tasks/idem_ports.yml
@@ -24,11 +24,16 @@
     image: "{{ idem_image }}"
     name: idempotency
     state: present
-    ports:
-      - 7777:8080/tcp
-      - 127.0.0.1:8888:1929
-      - 127.0.0.1:5555:9000
     command: 1h
+    ports:
+      - "4444:4444/tcp"
+      - "1212:5555"
+      - "8888:19191/udp"
+      - "1900:1900/udp"
+      - "127.0.0.1:7671:7676/udp"
+      - "127.0.0.1:12122:8876/udp"
+      - "127.0.0.1:13122:8871/tcp"
+      - "127.0.0.1:43423:8872"
   register: test2
 
 - name: check test2
@@ -41,9 +46,14 @@
     state: present
     command: 1h
     ports:
-      - 7777:8080/tcp
-      - 127.0.0.1:8888:1929
-      - 127.0.0.1:5555:9000
+      - "4444:4444/tcp"
+      - "1212:5555"
+      - "8888:19191/udp"
+      - "1900:1900/udp"
+      - "127.0.0.1:7671:7676/udp"
+      - "127.0.0.1:12122:8876/udp"
+      - "127.0.0.1:13122:8871/tcp"
+      - "127.0.0.1:43423:8872"
   register: test3
 
 - name: check test3