addressed part 2 feedback on unit test