bug fixing

Signed-off-by: goal101 <mark.kuhre@gmail.com>