diff --git a/components/code-block.tsx b/components/code-block.tsx
index c0780e5..336a1ca 100644
--- a/components/code-block.tsx
+++ b/components/code-block.tsx
@@ -1,20 +1,18 @@
 'use client';
 
 interface CodeBlockProps {
-  node: any;
-  inline: boolean;
-  className: string;
-  children: any;
+  node?: any;
+  className?: string;
+  children?: any;
 }
 
 export function CodeBlock({
   node,
-  inline,
   className,
   children,
   ...props
 }: CodeBlockProps) {
-  if (!inline) {
+  if (node?.children[0]?.value?.endsWith('\n')) {
     return (
       <div className="not-prose flex flex-col">
         <pre
diff --git a/components/markdown.tsx b/components/markdown.tsx
index 3c9ad9d..6fb9e5c 100644
--- a/components/markdown.tsx
+++ b/components/markdown.tsx
@@ -2,10 +2,10 @@ import Link from 'next/link';
 import React, { memo } from 'react';
 import ReactMarkdown, { type Components } from 'react-markdown';
 import remarkGfm from 'remark-gfm';
+
 import { CodeBlock } from './code-block';
 
 const components: Partial<Components> = {
-  // @ts-expect-error
   code: CodeBlock,
   pre: ({ children }) => <>{children}</>,
   ol: ({ node, children, ...props }) => {