minor fix on tests