small fix
some fixes