Fixed a bug in previous commit.

Signed-off-by: trungi <t.trungel@gmail.com>