Small fix
(cherry picked from commit 4dae78e)