diff --git a/docs/index.md b/docs/index.md
index c70f11e72..73e598af8 100644
--- a/docs/index.md
+++ b/docs/index.md
@@ -40,7 +40,7 @@ class Character(BaseModel):
     strength: int
 
 
-model = models.transformers("mistralai/Mistral-7B-v0.1", device="cuda")
+model = models.transformers("mistralai/Mistral-7B-v0.1")
 generator = generate.json(model, Character, max_tokens=100)
 sequence = generator("Give me a character description")
 ```
diff --git a/docs/stylesheets/extra.css b/docs/stylesheets/extra.css
index bbb46b81a..99671fd84 100644
--- a/docs/stylesheets/extra.css
+++ b/docs/stylesheets/extra.css
@@ -6,7 +6,7 @@
 
 .index-pre-code {
   margin: 50px;
-  width: 700px;
+  max-width: 700px;
   left: 50%;
 }