Minor bug fixed on tests.