diff --git a/packages/prism-react-renderer/CHANGELOG.md b/packages/prism-react-renderer/CHANGELOG.md
index 3ad138a..4055b8c 100644
--- a/packages/prism-react-renderer/CHANGELOG.md
+++ b/packages/prism-react-renderer/CHANGELOG.md
@@ -1,5 +1,12 @@
 # prism-react-renderer
 
+## 2.0.4
+
+### Patch Changes
+
+- Fix types for Prism library.
+  ([#204](https://github.com/FormidableLabs/prism-react-renderer/pull/204))
+
 ## 2.0.3
 
 ### Patch Changes
diff --git a/packages/prism-react-renderer/package.json b/packages/prism-react-renderer/package.json
index 1bdd1d9..0d9c9cd 100755
--- a/packages/prism-react-renderer/package.json
+++ b/packages/prism-react-renderer/package.json
@@ -1,6 +1,6 @@
 {
   "name": "prism-react-renderer",
-  "version": "2.0.3",
+  "version": "2.0.4",
   "description": "Renders highlighted Prism output using React",
   "sideEffects": true,
   "main": "dist/index.js",
diff --git a/packages/prism-react-renderer/src/types.ts b/packages/prism-react-renderer/src/types.ts
index 012a063..2e23120 100644
--- a/packages/prism-react-renderer/src/types.ts
+++ b/packages/prism-react-renderer/src/types.ts
@@ -1,25 +1,10 @@
 import type { CSSProperties } from "react"
-import type { Token as PrismToken } from "prismjs"
+import type { Token as PrismToken, Grammar } from "prismjs"
+import Prism from "prismjs"
 
 export type Language = string
-export type PrismGrammar = Record<string, unknown>
-type LanguagesDict = Record<Language, PrismGrammar>
-
-export type PrismLib = {
-  languages: LanguagesDict
-  tokenize: (code: string, grammar: PrismGrammar) => Array<PrismToken | string>
-  highlight: (code: string, grammar: PrismGrammar, language: Language) => string
-  hooks: {
-    run: (
-      name: string,
-      env: {
-        code: string
-        grammar: PrismGrammar
-        language: Language
-      }
-    ) => void
-  }
-}
+export type PrismGrammar = Grammar
+export type PrismLib = typeof Prism
 
 export type Token = {
   types: string[]