small fixes
small fix