Fixed error on tests