fixed bug.

Signed-off-by: Jonathan <lovattj@me.com>