fixed bug on tests