small tweak to tests