diff --git a/deps/rabbitmq_cli/test/diagnostics/memory_breakdown_command_test.exs b/deps/rabbitmq_cli/test/diagnostics/memory_breakdown_command_test.exs
index d010135ea6..e7997de4a2 100644
--- a/deps/rabbitmq_cli/test/diagnostics/memory_breakdown_command_test.exs
+++ b/deps/rabbitmq_cli/test/diagnostics/memory_breakdown_command_test.exs
@@ -53,11 +53,11 @@ defmodule MemoryBreakdownCommandTest do
   end
 
   test "validate: specifying gigabytes as a --unit succeeds", context do
-    assert @command.validate([], Map.merge(context[:opts], %{unit: "mb"})) == :ok
+    assert @command.validate([], Map.merge(context[:opts], %{unit: "gb"})) == :ok
   end
 
   test "validate: specifying bytes as a --unit succeeds", context do
-    assert @command.validate([], Map.merge(context[:opts], %{unit: "mb"})) == :ok
+    assert @command.validate([], Map.merge(context[:opts], %{unit: "bytes"})) == :ok
   end
 
   test "validate: specifying megabytes as a --unit succeeds", context do