fix for issue
small fixes