small modification on tests.