diff --git a/tests/yaml_nvim/commands_spec.lua b/tests/yaml_nvim/commands_spec.lua
index 0ae0028..445d35b 100644
--- a/tests/yaml_nvim/commands_spec.lua
+++ b/tests/yaml_nvim/commands_spec.lua
@@ -16,4 +16,24 @@ describe("commands", function()
 
 		assert(vim.fn.getreg("7"), "worldcup")
 	end)
+
+	it("yanks value to the default register", function()
+		require("yaml_nvim")
+
+		vim.cmd(":e sample.yaml")
+		vim.cmd("norm 10j")
+		vim.cmd(":YAMLYankValue")
+
+		assert(vim.fn.getreg('"'), "7")
+	end)
+
+	it("yanks value to a custom register", function()
+		require("yaml_nvim")
+
+		vim.cmd(":e sample.yaml")
+		vim.cmd("norm 10j")
+		vim.cmd(":YAMLYankValue 7")
+
+		assert(vim.fn.getreg("7"), "7")
+	end)
 end)