Updating unit tests to match module.: