minor tweaks on test