1 files changed, 2 insertions, 2 deletions
diff --git a/spec/frontend/notebook/lib/highlight_spec.js b/spec/frontend/notebook/lib/highlight_spec.js
index d71c5718858..944ccd6aa9f 100644
--- a/spec/frontend/notebook/lib/highlight_spec.js
+++ b/spec/frontend/notebook/lib/highlight_spec.js
@@ -9,7 +9,7 @@ describe('Highlight library', () => {
     const el = document.createElement('div');
     el.innerHTML = Prism.highlight('console.log("a");', Prism.languages.javascript);
 
-    expect(el.querySelector('.s')).not.toBeNull();
-    expect(el.querySelector('.nf')).not.toBeNull();
+    expect(el.querySelector('.string')).not.toBeNull();
+    expect(el.querySelector('.function')).not.toBeNull();
   });
 });