fixed bug

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