Fixed mistake on tests.