1 files changed, 18 insertions, 25 deletions
diff --git a/tests/language-feature/capability/capability3.slang b/tests/language-feature/capability/capability3.slang
index 02eb7d495..96c07a51f 100644
--- a/tests/language-feature/capability/capability3.slang
+++ b/tests/language-feature/capability/capability3.slang
@@ -3,47 +3,40 @@
 // CHECK_IGNORE_CAPS-NOT: error 36108
 
 // Test that capabilities can be declared on module.
-
 [require(glsl)]
 [require(spirv)]
 module test;
 
-void f()
-{
-    __require_capability glsl;
-}
-
-// CHECK: ([[# @LINE+1]]): error 36108
-public void g()
-{
-    __require_capability spvAtomicFloat16AddEXT;
-}
+RWStructuredBuffer<int> sideEffect;
 
-void l()
+// CHECK: error 36104
+[require(glsl, _sm_4_0)]
+public void use1()
 {
     __target_switch
     {
     case glsl:
-        f();
-        return;
-    case spirv:
-        __require_capability spvAtomicFloat16AddEXT;
         return;
     }
 }
 
-// CHECK: ([[# @LINE+1]]): error 36104: {{.*}}
-public void use()
+void use2Sub()
 {
-    l(); // Error
+    __target_switch
+    {
+    case glsl:
+        sideEffect[1] = 1;
+    }
 }
-
-// CHECK-NOT: ([[# @LINE+1]]): error
-[require(spirv, spvAtomicFloat16AddEXT)]
-public void use1()
+// CHECK: error 36108
+[require(spirv, spirv_1_0)]
+public void use2()
 {
-    l(); // Error
+    use2Sub();
 }
 
 void main()
-{}
-\ No newline at end of file
+{
+    use1();
+    use2();
+}
+\ No newline at end of file