small fix
small changes