few fixes
small fix