minor changes to tests