diff --git a/spec/javascripts/blob/notebook/index_spec.js b/spec/javascripts/blob/notebook/index_spec.js
index 03539bead29..11f2a950678 100644
--- a/spec/javascripts/blob/notebook/index_spec.js
+++ b/spec/javascripts/blob/notebook/index_spec.js
@@ -76,11 +76,11 @@ describe('iPython notebook renderer', () => {
 
     it('highlights code', () => {
       expect(
-        document.querySelector('.hljs'),
+        document.querySelector('.token'),
       ).not.toBeNull();
 
       expect(
-        document.querySelector('.python'),
+        document.querySelector('.language-python'),
       ).not.toBeNull();
     });
   });