minor changes in test classes