temporary fix on tests