diff --git a/google/generativeai/__init__.py b/google/generativeai/__init__.py
index 821d8aed7..6b74f83a8 100644
--- a/google/generativeai/__init__.py
+++ b/google/generativeai/__init__.py
@@ -22,6 +22,7 @@
 
 ```
 import google.generativeai as palm
+import os
 
 palm.configure(api_key=os.environ['API_KEY'])
 ```