diff --git a/cli/command/formatter/container_test.go b/cli/command/formatter/container_test.go
index 96626d29f0..b5a96950c1 100644
--- a/cli/command/formatter/container_test.go
+++ b/cli/command/formatter/container_test.go
@@ -54,14 +54,51 @@ func TestContainerPsContext(t *testing.T) {
 			call:      ctx.Image,
 		},
 		{
-			container: container.Summary{Image: "verylongimagename"},
+			container: container.Summary{Image: "ubuntu:latest"},
 			trunc:     true,
-			expValue:  "verylongimagename",
+			expValue:  "ubuntu:latest",
 			call:      ctx.Image,
 		},
 		{
-			container: container.Summary{Image: "verylongimagename"},
-			expValue:  "verylongimagename",
+			container: container.Summary{Image: "docker.io/library/ubuntu"},
+			trunc:     true,
+			expValue:  "ubuntu",
+			call:      ctx.Image,
+		},
+		{
+			container: container.Summary{Image: "docker.io/library/ubuntu:latest"},
+			trunc:     true,
+			expValue:  "ubuntu:latest",
+			call:      ctx.Image,
+		},
+		{
+			container: container.Summary{Image: "ubuntu:latest@sha256:a5a665ff33eced1e0803148700880edab4269067ed77e27737a708d0d293fbf5"},
+			trunc:     true,
+			expValue:  "ubuntu:latest",
+			call:      ctx.Image,
+		},
+		{
+			container: container.Summary{Image: "ubuntu@sha256:a5a665ff33eced1e0803148700880edab4269067ed77e27737a708d0d293fbf5"},
+			trunc:     true,
+			expValue:  "ubuntu",
+			call:      ctx.Image,
+		},
+		{
+			container: container.Summary{Image: "docker.io/library/ubuntu@sha256:a5a665ff33eced1e0803148700880edab4269067ed77e27737a708d0d293fbf5"},
+			trunc:     true,
+			expValue:  "ubuntu",
+			call:      ctx.Image,
+		},
+		{
+			container: container.Summary{Image: "docker.io/library/ubuntu:latest@sha256:a5a665ff33eced1e0803148700880edab4269067ed77e27737a708d0d293fbf5"},
+			trunc:     true,
+			expValue:  "ubuntu:latest",
+			call:      ctx.Image,
+		},
+		{
+			container: container.Summary{Image: "verylongimagenameverylongimagenameverylongimagenameverylongimagenameverylongimagenameverylongimagenameverylongimagename"},
+			trunc:     true,
+			expValue:  "verylongimagenameverylongimagenameverylongimagenameverylongimagenameverylongimagenameverylongimagenameverylongimagename",
 			call:      ctx.Image,
 		},
 		{