diff --git a/spec/lib/rufo/erb_formatter_spec.rb b/spec/lib/rufo/erb_formatter_spec.rb
index 9ad8816c..8db16d52 100644
--- a/spec/lib/rufo/erb_formatter_spec.rb
+++ b/spec/lib/rufo/erb_formatter_spec.rb
@@ -90,6 +90,16 @@
       expect(result).to eql("<%= yield %>")
     end
 
+    it "formats standalone 'yield' with arguments" do
+      result = subject.format("<%=yield x,y%>")
+      expect(result).to eql("<%= yield x, y %>")
+    end
+
+    it "formats standalone 'yield' with arguments and parens" do
+      result = subject.format("<%=yield(x,y)%>")
+      expect(result).to eql("<%= yield(x, y) %>")
+    end
+
     it "handles native erb comments" do
       result = subject.format("<%# locals: (item:, variant:) %>")
       expect(result).to eql("<%# locals: (item:, variant:) %>")
@@ -99,5 +109,20 @@
       result = subject.format("<% # TODO: fix this later %>")
       expect(result).to eql("<% # TODO: fix this later %>")
     end
+
+    it "handles case/when expression" do
+      result = subject.format(<<~ERB)
+        <% case a+b %>
+        <% when c %>
+        <%= d+e %>
+        <% end %>
+      ERB
+      expect(result).to eql(<<~ERB)
+                          <% case a + b %>
+                          <% when c %>
+                          <%= d + e %>
+                          <% end %>
+                        ERB
+    end
   end
 end