Minor bug fixed.

Signed-off-by: Justin Lee <lis186@gmail.com>