Minor bug re-fixed on tests.