diff --git a/burn-book/src/basic-workflow/model.md b/burn-book/src/basic-workflow/model.md
index 6f33a4d22..90092a4e8 100644
--- a/burn-book/src/basic-workflow/model.md
+++ b/burn-book/src/basic-workflow/model.md
@@ -10,7 +10,7 @@ cargo new my-first-burn-model
 As [mentioned previously](../getting-started.md#creating-a-burn-application), this will initialize
 your `my-first-burn-model` project directory with a `Cargo.toml` and a `src/main.rs` file.
 
-In the `Cargo.toml` file, add the `burn` dependency with `train` and `wgpu` features.
+In the `Cargo.toml` file, add the `burn` dependency with `train`, `wgpu` and `vision` features.
 
 ```toml
 [package]
@@ -19,7 +19,7 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
-burn = { version = "0.13.0", features = ["train", "wgpu"] }
+burn = { version = "0.13.0", features = ["train", "wgpu", "vision"] }
 ```
 
 Our goal will be to create a basic convolutional neural network used for image classification. We