diff --git a/docs/app/app.ts b/docs/app/app.ts
index a6f13acd..beadc625 100644
--- a/docs/app/app.ts
+++ b/docs/app/app.ts
@@ -9,6 +9,7 @@ import 'prismjs/components/prism-scss';
 import 'prismjs/components/prism-javascript';
 import 'prismjs/components/prism-handlebars';
 import 'prismjs/components/prism-markup-templating';
+// @ts-expect-error no types shipped from prismjs-glimmer
 import { setup } from 'prismjs-glimmer';
 
 import 'prismjs/themes/prism.css';
diff --git a/docs/tests/integration/components/code-block-test.ts b/docs/tests/integration/components/code-block-test.ts
index ebb7f7ef..2f04ff7c 100644
--- a/docs/tests/integration/components/code-block-test.ts
+++ b/docs/tests/integration/components/code-block-test.ts
@@ -10,17 +10,8 @@ module('Integration | Component | code-block', function (hooks) {
     // Set any properties with this.set('myProperty', 'value');
     // Handle any actions with this.set('myAction', function(val) { ... });
 
-    await render(hbs`<CodeBlock />`);
+    await render(hbs`<CodeBlock @language="js" @code="console.log('hello');" />`);
 
-    assert.dom().hasText('');
-
-    // Template block usage:
-    await render(hbs`
-      <CodeBlock>
-        template block text
-      </CodeBlock>
-    `);
-
-    assert.dom().hasText('template block text');
+    assert.dom().hasText('console.log(\'hello\');');
   });
 });
diff --git a/docs/tests/integration/components/code-inline-test.ts b/docs/tests/integration/components/code-inline-test.ts
index a1144a04..ad480277 100644
--- a/docs/tests/integration/components/code-inline-test.ts
+++ b/docs/tests/integration/components/code-inline-test.ts
@@ -10,17 +10,8 @@ module('Integration | Component | code-inline', function (hooks) {
     // Set any properties with this.set('myProperty', 'value');
     // Handle any actions with this.set('myAction', function(val) { ... });
 
-    await render(hbs`<CodeInline />`);
+    await render(hbs`<CodeInline @language="js" @code="console.log('hello');" />`);
 
-    assert.dom().hasText('');
-
-    // Template block usage:
-    await render(hbs`
-      <CodeInline>
-        template block text
-      </CodeInline>
-    `);
-
-    assert.dom().hasText('template block text');
+    assert.dom().hasText('console.log(\'hello\');');
   });
 });