small fix on tests