small changes on tests
small changes on tests