diff --git a/docs/my-website/src/components/queryParamReader.js b/docs/my-website/src/components/queryParamReader.js
index d6bc2d740ca..9a5e50502b3 100644
--- a/docs/my-website/src/components/queryParamReader.js
+++ b/docs/my-website/src/components/queryParamReader.js
@@ -2,6 +2,7 @@ import React, { useState, useEffect } from 'react';
 
 const CodeBlock = ({ token }) => {
   const codeWithToken = `
+import os
 from litellm import completion
 
 # set ENV variables