fixed bug...

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