diff --git a/test/cli/prune_test.rb b/test/cli/prune_test.rb
index 15e54575..4ff9eedc 100644
--- a/test/cli/prune_test.rb
+++ b/test/cli/prune_test.rb
@@ -10,15 +10,13 @@ class CliPruneTest < CliTestCase
 
   test "images" do
     run_command("images").tap do |output|
-      assert_match "docker image prune --all --force --filter label=service=app --filter until=168h on 1.1.1.1", output
-      assert_match "docker image prune --all --force --filter label=service=app --filter until=168h on 1.1.1.2", output
+      assert_match /docker image prune --all --force --filter label=service=app --filter until=168h on 1.1.1.\d/, output
     end
   end
 
   test "containers" do
     run_command("containers").tap do |output|
-      assert_match "docker container prune --force --filter label=service=app --filter until=72h on 1.1.1.1", output
-      assert_match "docker container prune --force --filter label=service=app --filter until=72h on 1.1.1.2", output
+      assert_match /docker container prune --force --filter label=service=app --filter until=72h on 1.1.1.\d/, output
     end
   end
 
diff --git a/test/cli/registry_test.rb b/test/cli/registry_test.rb
index d100402c..0647f873 100644
--- a/test/cli/registry_test.rb
+++ b/test/cli/registry_test.rb
@@ -4,15 +4,13 @@ class CliRegistryTest < CliTestCase
   test "login" do
     run_command("login").tap do |output|
       assert_match /docker login -u \[REDACTED\] -p \[REDACTED\] as .*@localhost/, output
-      assert_match "docker login -u [REDACTED] -p [REDACTED] on 1.1.1.1", output
-      assert_match "docker login -u [REDACTED] -p [REDACTED] on 1.1.1.2", output
+      assert_match /docker login -u \[REDACTED\] -p \[REDACTED\] on 1.1.1.\d/, output
     end
   end
 
   test "logout" do
     run_command("logout").tap do |output|
-      assert_match "docker logout on 1.1.1.1", output
-      assert_match "docker logout on 1.1.1.2", output
+      assert_match /docker logout on 1.1.1.\d/, output
     end
   end