diff --git a/cli/src/tests/tags_test.rs b/cli/src/tests/tags_test.rs
index d293806b..f8ad2527 100644
--- a/cli/src/tests/tags_test.rs
+++ b/cli/src/tests/tags_test.rs
@@ -9,10 +9,14 @@ fn test_tags_javascript() {
         r#"
         ((function_definition
             name: (identifier) @name
-            body: (block . (string) @doc)) @function
+            body: (block . (expression_statement (string) @doc))) @function
          (set! replace @doc "(^['\s]*)|(['\s]*$)"))
         (function_definition
             name: (identifier) @name) @function
+        ((class_definition
+            name: (identifier) @name
+            body: (block . (expression_statement (string) @doc))) @class
+         (set! replace @doc "(^['\s]*)|(['\s]*$)"))
         (class_definition
             name: (identifier) @name) @class
         (call