diff --git a/README.md b/README.md
index 035827992..ab1dc9b32 100644
--- a/README.md
+++ b/README.md
@@ -40,8 +40,8 @@ struct Model: Layer {
 #### Initialize a model and an optimizer
 
 ```swift
-let optimizer = SGD<Model, Float>(learningRate: 0.02)
 var classifier = Model()
+let optimizer = SGD(learningRate: 0.02, for: classifier)
 let context = Context(learningPhase: .training)
 let x: Tensor<Float> = ...
 let y: Tensor<Float> = ...