Updating unit tests with feedback from review