small changes in test