removing return in unit tests.