small changes in tests