Little changes in test