fix
small tweaks