small fix
small fixes