diff --git a/README.md b/README.md
index 034eccb..4e47f56 100644
--- a/README.md
+++ b/README.md
@@ -137,11 +137,7 @@ You can use context managers to temporarily enforce the maintenance state for sp
 
 ```python
 from fastapi import FastAPI
-from fastapi_maintenance import (
-    MaintenanceModeMiddleware,
-    maintenance_mode_off,
-    maintenance_mode_on,
-)
+from fastapi_maintenance import MaintenanceModeMiddleware, maintenance_mode_on
 
 app = FastAPI()
 app.add_middleware(MaintenanceModeMiddleware)
diff --git a/docs/index.md b/docs/index.md
index d7180ab..eeea615 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -137,11 +137,7 @@ You can use context managers to temporarily enforce the maintenance state for sp
 
 ```python
 from fastapi import FastAPI
-from fastapi_maintenance import (
-    MaintenanceModeMiddleware,
-    maintenance_mode_off,
-    maintenance_mode_on,
-)
+from fastapi_maintenance import MaintenanceModeMiddleware, maintenance_mode_on
 
 app = FastAPI()
 app.add_middleware(MaintenanceModeMiddleware)