Added a couple unit tests to make sure we have consistent coverage on this function