diff --git a/src/functor.jl b/src/functor.jl
index ba8c9212..0d7c55f1 100644
--- a/src/functor.jl
+++ b/src/functor.jl
@@ -46,7 +46,7 @@ Set a layer or model's test mode (see below).
 Using `:auto` mode will treat any gradient computation as training.
 
 _Note_: if you manually set a model into test mode, you need to manually place
-it back into train mode.
+it back into train mode during training phase.
 
 Possible values include:
 - `false` for training
@@ -62,7 +62,7 @@ Set a layer of model's train mode (see below).
 Symmetric to [`testmode!`](@ref) (i.e. `trainmode!(m, mode) == testmode!(m, !mode)).
 
 _Note_: if you manually set a model into train mode, you need to manually place
-it into test mode.
+it into test mode during testing phase.
 
 Possible values include:
 - `true` for training