small changes to tests