minor update on tests