Fix code.

(cherry picked from commit 687d8fbc66fb248d36e8eedf0e19f1990dc1ca71)