diff --git a/bert/README.md b/bert/README.md
index 42e5e957..e03df7d0 100644
--- a/bert/README.md
+++ b/bert/README.md
@@ -23,6 +23,7 @@ python convert.py \
 To use the `Bert` model in your own code, you can load it with:
 
 ```python
+import mlx.core as mx
 from model import Bert, load_model
 
 model, tokenizer = load_model(
diff --git a/stable_diffusion/README.md b/stable_diffusion/README.md
index 385ce6d9..711ec10c 100644
--- a/stable_diffusion/README.md
+++ b/stable_diffusion/README.md
@@ -31,6 +31,7 @@ way to get started is by using the `StableDiffusion` class from the `stable_diff
 module.
 
 ```python
+import mlx.core as mx
 from stable_diffusion import StableDiffusion
 
 # This will download all the weights from HF hub and load the models in