diff --git a/README.md b/README.md
index f2efaa0..26d509a 100644
--- a/README.md
+++ b/README.md
@@ -2319,7 +2319,8 @@ async def view(state, screen):
             screen.addstr(p.y, p.x, str(id_))
         await asyncio.sleep(0.01)  
 
-curses.wrapper(main)
+if __name__ == '__main__':
+    curses.wrapper(main)
 ```
 <br>
 
diff --git a/index.html b/index.html
index 9a75eb7..d9263e3 100644
--- a/index.html
+++ b/index.html
@@ -2042,7 +2042,8 @@ D = enum.Enum(<span class="hljs-string">'D'</span>, <span class="hljs-string">'n
             screen.addstr(p.y, p.x, str(id_))
         <span class="hljs-keyword">await</span> asyncio.sleep(<span class="hljs-number">0.01</span>)  
 
-curses.wrapper(main)
+<span class="hljs-keyword">if</span> __name__ == <span class="hljs-string">'__main__'</span>:
+    curses.wrapper(main)
 </code></pre></div></div>