Small fix
(cherry picked from commit 9f67904)